描述
n*m的矩阵,k个点,将k个点全部放在n*m的矩阵里,求满足以下约束的方案数:
矩阵第一行,第一列,最后一行,最后一列都有点。
输出方案数对1e9+7的模数
示例1
输入:
2,3,1
返回值:
0
说明:
就1个点,所以无法满足条件。
示例2
输入:
2,2,2
返回值:
2
说明:
我们可以把1个点放在左上角,第一个点放在右下角;也可以把一个点放在右上角,一个点放在左下角。故而有2种放法。
备注:
2<=n,m<=30; 1<=k<=1000
以下是Java代码实现:
public class Main {
public static int countSolutions(int n, int m, int k) {
if (k < 4) {
return 0;
}
int mod = 1000000007;
int[][][] dp = new int[n][m][k];
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
if (i == 0 || j == 0 || i == n - 1 || j == m - 1) {
dp[i][j][1] = 1;
}
}
}
for (int i = 2; i <= k; i++) {
for (int j = 0; j < n; j++) {
for (int l = 0; l < m; l++) {
for (int p = 0; p < n; p++) {
for (int q = 0; q < m; q++) {
if (j == p || l == q) {
continue;
}
dp[j][l][i] = (dp[j][l][i] + dp[p][q][i - 1]) % mod;
}
}
}
}
}
int ans = 0;
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
ans = (ans + dp[i][j][k]) % mod;
}
}
return ans;
}
public static void main(String[] args) {
int n = 2;
int m = 3;
int k = 1;
int result = countSolutions(n, m, k);
System.out.println(result);
}
}
代码解释:
- 定义一个countSolutions()函数,用于计算方案数。
- 如果k小于4,则无法满足条件,返回0。
- 定义一个dp数组,用于记录每个位置和点数的方案数。
- 初始化dp数组,将第一行、第一列、最后一行、最后一列的方案数设为1。
- 使用四层循环,遍历dp数组,更新dp数组的值。
- 对于每个位置和点数,尝试执行以下操作:
- 将一个点放在当前位置。
- 将另一个点放在其他位置。
- 将剩余的点放在其他位置。
- 返回总方案数。
- 在main()函数中,调用countSolutions()函数,并输出结果。