LeetCode LCP 22. 黑白方格画（二进制枚举）

题意：

小扣注意到秋日市集上有一个创作黑白方格画的摊位。
摊主给每个顾客提供一个固定在墙上的白色画板，画板不能转动。
画板上有 n * n 的网格。绘画规则为，小扣可以选择任意多行以及任意多列的格子涂成黑色，所选行数、列数均可为 0。

小扣希望最终的成品上需要有 k 个黑色格子，请返回小扣共有多少种涂色方案。

注意：两个方案中任意一个相同位置的格子颜色不同，就视为不同的方案。

数据范围:
1 <= n <= 6
0 <= k <= n * n

解法：

观察到n比较小,可以用二进制枚举枚举选择的行和列的方案数,
如果方案的黑色数量=k则ans++.

注意当k=n*n的时候,需要特判返回1,否则会wa.

code：

class Solution {
public:
    int paintingPlan(int n, int k) {
        if(k==n*n)return 1;
        int ans=0;
        for(int i=0;i<(1<<n);i++){
            for(int j=0;j<(1<<n);j++){
                int cnt1=0,cnt2=0;
                for(int k=0;k<n;k++){
                    if(i>>k&1)cnt1++;//选择当前行
                    if(j>>k&1)cnt2++;//选择当前列
                }
                int cnt=(cnt1+cnt2)*n-cnt1*cnt2;
                if(cnt==k)ans++;
            }
        }
        return ans;
    }
};

---