python - 获取满足条件的所有可能的 3x2 矩阵的数量

Question

我想得到所有可能的 3x2 矩阵的数量

• 矩阵的每个元素都是 0、1 或 2。
• 矩阵的 3 行在所有条目中都没有相同的元素(a_{i1} != a_{i2} for i=1,2,3)。
• 矩阵的 2 列中没有一列在他的所有条目中具有相同的元素(a_{1j} = a_{2j} = a_{3j} 永远不会是 j=1,2 的情况)。

基本上，如果 0、1 和 2 由颜色 0 = 红色、1 = 绿色、2 = 蓝色表示，那么您可以生成以下矩阵:

禁止矩阵配置:

如何使用此配置获取所有可能矩阵的数量？或者我如何生成它们？

Answer 1

您可以使用递归和回溯生成。每次您将尝试用三种颜色中的一种填充其中一个单元格，然后调用下一个单元格。如果最终所有颜色都填满，则检查矩阵是否有效。

def recursion(Matrix ,i ,j):
    # Validation check
    if i==3:
        if  Matrix[0][0]==Matrix[1][0] and Matrix[1][0]==Matrix[2][0]:
            return;
        if  Matrix[0][1]==Matrix[1][1] and Matrix[1][1]==Matrix[2][1]:
            return;
        if  Matrix[0][0]==Matrix[0][1] or Matrix[1][0]==Matrix[1][1] or Matrix[2][0]==Matrix[2][1]:
            return
        print("Matrix : ")
        for row in Matrix:
            for val in row:
                print '{:4}'.format(val),
            print
        return

    # Shifting to the next row
    if j==2:
        recursion(Matrix,i+1,0)
        return

    # Filling up the current cell by one of {0,1,2}
    for color in range(3):
        Matrix[i][j]=color
        recursion(Matrix,i,j+1)
    return


Matrix=[[0 for x in range(2)] for y in range(3)]
recursion(Matrix,0,0)