leecode-1434

题目链接

题目

思路分析

这道题题目中,人数是小于等于10的,也就意味着,每个人的状态用0/1来代表的时候,是不会爆int的.因此我们用一个比特长度为10的数字来代表 ,如果全部为1的情况下,证明大家都是戴上了帽子.因此就用一个dp状态压缩来做,代码如下.

代码

class Solution {
public:
    typedef long long ll; //声明long long类型,因为爆int
    const int MOD = 1e9 + 7; //声明求余的变量
    int numberWays(vector<vector<int>>& hats) {
        int n = hats.size();
        int tState = 1 << n;
        vector<ll> dp(tState, 0);
        dp[0] = 1;
        vector<set<int>> person(41);
        for(int i = 0; i < n;i++){
            for(int h : hats[i]){
                person[h].insert(i);
            }
        }

        for(int i = 1 ;i <= 40;i++){
            for(int state = tState - 1; state >= 0; state--){
                for(int p : person[i]){
                    if(!(state & (1 << p))){//如果当前这个人还没有帽子的话,就先分给这个人,然后遍历状态
                        dp[state + (1 << p)] += dp[state];
                        dp[state + (1 << p)] %= MOD;
                    }
                }
            }
        }
        return dp[tState - 1];
    }
};