LCP 22. 黑白方格画

/**
 * LCP 22. 黑白方格画
 * @author wsq
 * @date 2020/10/18
 	小扣注意到秋日市集上有一个创作黑白方格画的摊位。摊主给每个顾客提供一个固定在墙上的白色画板，画板不能转动。画板上有 n * n 的网格。绘画规则为，小扣可以选择任意多行以及任意多列的格子涂成黑色，所选行数、列数均可为 0。
	小扣希望最终的成品上需要有 k 个黑色格子，请返回小扣共有多少种涂色方案。
	注意：两个方案中任意一个相同位置的格子颜色不同，就视为不同的方案。

	示例 1：
	输入：n = 2, k = 2
	输出：4
	解释：一共有四种不同的方案：
	第一种方案：涂第一列；
	第二种方案：涂第二列；
	第三种方案：涂第一行；
	第四种方案：涂第二行。
	
	链接：https://leetcode-cn.com/problems/ccw6C7
 */
package com.wsq.leetcode;

public class PaintingPlan {
	/**
	 *	1.从选择多少条行和多少条列出发，寻找满足条件的行数和列数
	 *	2.计算出满足条件的（行，列）组合有多少种情况
	 * @param n
	 * @param k
	 * @return
	 */
	public int paintingPlan(int n, int k) {
        if(n*n == k || k == 0){
            return 1;
        }
        if(k < n){
            return 0;
        }
        int ans = 0;
        for(int i = 0; i <= n; i++){
            for(int j = 0; j <= n; j++){
                if(n*(i + j) - (j * i) == k){
                    ans += get(n, i) * get(n, j);
                }
            }
        }
        return ans;
    }
    
    private int get(int n, int a){
        int res = 1;
        for(int i = n; i > n-a; i--){
            res *= i;
        }
        for(int i = 1; i <= a; i++){
            res /= i;
        }
        return res;
    }
}