求助这种代码怎么构建啊

A,B两个矩阵。大小任意，举个例子都是6*6的方阵。并且A B两个矩阵都只有一个矩阵元并且在对角元上，对于A矩阵比如(1,1)=1其他都是0，那么B矩阵同理，就是除了(1,1)以外的一个位置有值等于1，例如在(2,2)或者(3,3)等等。那么现在对于A矩阵和B矩阵求乘积，就有30种情况(A在1的话，B就在2 3 4 5 6。 A在2的话，B就在1 3 4 5 6 同理)的乘积。 有没有大佬可以给点思路，这样的程序应该怎么写啊。