我正在尝试找到一种算法(不是 matlab 命令)来枚举所有可能的 NxM 矩阵,其约束条件是每个单元格(或 0)中只有正整数和每行和列的固定总和(这些是参数算法)。
示例:枚举所有行总数为 2、1 和列总数为 0、1、2 的 2x3 矩阵:
| 0 0 2 | = 2
| 0 1 0 | = 1
0 1 2
| 0 1 1 | = 2
| 0 0 1 | = 1
0 1 2
这是一个相当简单的例子,但随着 N 和 M 以及总和的增加,可能会有很多可能性。
编辑 1
我可能有一个有效的安排来启动算法:
matrix = new Matrix(N, M) // NxM matrix filled with 0s
FOR i FROM 0 TO matrix.rows().count()
FOR j FROM 0 TO matrix.columns().count()
a = target_row_sum[i] - matrix.rows[i].sum()
b = target_column_sum[j] - matrix.columns[j].sum()
matrix[i, j] = min(a, b)
END FOR
END FOR
target_row_sum[i] 是第 i 行的预期总和。
在上面的例子中,它给出了第二种排列方式。
编辑 2:(基于j_random_hacker 的最后一条语句)
让 M 是验证给定条件的任何矩阵(行和列总和固定、正或空单元格值)。令 (a, b, c, d) 为 M 中的 4 个单元格值,其中 (a, b) 和 (c, d) 在同一行上,并且 (a, c) 和 (b, d) 在同一行上柱子。令 Xa 为包含 a 的单元格的行号,Ya 为其列号。
例子:
| 1 a b |
| 1 2 3 |
| 1 c d |
-> Xa = 0, Ya = 1
-> Xb = 0, Yb = 2
-> Xc = 2, Yc = 1
-> Xd = 2, Yd = 2
这是一种算法,可以让所有组合验证初始条件并仅使 a、b、c 和 d 变化:
// A matrix array containing a single element, M
// It will be filled with all possible combinations
matrices = [M]
I = min(a, d)
J = min(b, c)
FOR i FROM 1 TO I
tmp_matrix = M
tmp_matrix[Xa, Ya] = a - i
tmp_matrix[Xb, Yb] = b + i
tmp_matrix[Xc, Yc] = c - i
tmp_matrix[Xd, Yd] = d + i
matrices.add(tmp_matrix)
END FOR
FOR j FROM 1 TO J
tmp_matrix = M
tmp_matrix[Xa, Ya] = a + j
tmp_matrix[Xb, Yb] = b - j
tmp_matrix[Xc, Yc] = c + j
tmp_matrix[Xd, Yd] = d - j
matrices.add(tmp_matrix)
END FOR
然后应该可以找到矩阵值的每个可能组合:
- 将算法应用于每个可能的 4 个单元组的第一个矩阵;
- 递归地将该算法应用于先前迭代获得的每个子矩阵,对于每个可能的 4 个单元组,除了已在父执行中使用的任何组;
递归深度应该是(N*(N-1)/2)*(M*(M-1)/2)
,每次执行都会产生((N*(N-1)/2)*(M*(M-1)/2) - depth)*(I+J+1)
子矩阵。但这会产生很多重复的矩阵,所以这可能会被优化。