leetcode黑白方格画
题目截图:
题目限制:
示例 1:
输入:n = 2, k = 2
输出:4
解释:一共有四种不同的方案:
第一种方案:涂第一列;
第二种方案:涂第二列;
第三种方案:涂第一行;
第四种方案:涂第二行。
示例 2:
输入:n = 2, k = 1
输出:0
解释:不可行,因为第一次涂色至少会涂两个黑格。
示例 3:
输入:n = 2, k = 4
输出:1
解释:共有 2*2=4 个格子,仅有一种涂色方案。
限制:
1 <= n <= 6
0 <= k <= n * n
这道题简单,此处主要是记录一个排列组合公式,以便日后提醒自己。
这道题,就是从n*n个方格中找到k个方格(保证满行或满列)即可:
问题可分解为两个排列组合:从n行找出i行 和 从n列找出j列;
if(这个两个组合可以得到k个方格){
再把这两个的组合数目相乘,即为挑选i行j列的所有情况
}
此时,只要把所有(挑0行到挑n行,挑0列到挑n列)符合题意的情况求和即可
class Solution {
public int paintingPlan(int n, int k) {
if(k==0||n*n==k){
return 1;
}else if(k<n){
return 0;
}else{
int sum_num=0;
for(int i=0;i<=n;i++){
for(int j=0;j<=n;j++){
if((i+j)*n-i*j==k){
//从n行里面挑i行,从n列里挑j列,二者情况相乘
sum_num+=(jc(n)/jc(i)/jc(n-i))*(jc(n)/jc(j)/jc(n-j)) ;
}
}
}
return sum_num;
}
}
public int jc(int n){
if(n<=1)
return 1;
return n*jc(n-1);
}
}
利用排列组合的两个公式:
就可以得到一个递推公式:
C[n - 1][m - 1] + C[n - 1][m] = C[n][m]
此时,就不在需要用阶乘相除
求C[n][m],用循环也可以求出:
for(int i = 0; i <= n; i++){
for(int j = 0; j <= i/2+1; j++){
if(!j) C[i][j] = 1;//从i中挑0个,仅有一种情况,即C[n][0]=1;
else C[i][j] = C[i - 1][j - 1] + C[i - 1][j];
}
//下面这个循环不写也行,就是用的时候需要写成C[i][i-j]即可
for(int j = i-i/2+1; j <= i; j++)
C[i][j]=C[i][i-j];
}