[leetcode 25场周赛第四题] 每个人戴不同帽子的方案数

呜呜，昨晚做的时候用的深搜回溯，发现超时了，没做出来
今天参考大佬的写法记录一下

题目

https://leetcode-cn.com/problems/number-of-ways-to-wear-different-hats-to-each-other/

分析

一开始看了看数据，只想到深搜回溯。。。想着能不能偷个鸡，结果后台数据还是挺给力的。

看了榜首大佬的写法，是一个状态压缩的动态规划问题，压缩方式还是以二进制的形式进行压缩。

这样的好处就是把原本40* 40的十次方的复杂度可以降低到10 * 1024 * 40

dfs深搜（超时）

class Solution {
public:
    int ans = 0;
    const int mod = 1000000007;
    int len = 0;
    int numberWays(vector<vector<int>>& hats) {
        vector<int> h(50,0);
        len = hats.size();
        dfs(h,hats,0);
        return ans;
    }
    
    void dfs(vector<int>& h, vector<vector<int>>& hats,int index) {
    	if(index >= len) 
    	{
    		ans = (ans + 1) % mod;
    		return ;
    	}
    	
        for(int i = 0; i < hats[index].size(); i++)
        {
            int tmp = hats[index][i];
            if(!h[tmp])
            {
                h[tmp] = 1;
                dfs(h,hats,index+1);
                h[tmp] = 0;
            }
        }
    }
};

dp状态压缩

• dp[i][j]前 i 个帽子，所有人的状态为 j 时满足的方案数
• 一共有10个人，每个人的状态就可以用 一个只有一个1的十位的二进制来表示 2的十次方 = 1024

class Solution {
public:
    const int mod = 1000000007;
    int numberWays(vector<vector<int>>& hats) {
        int len = hats.size();//一共有len个人
        int maxhats = 0;
        //找一个帽子的最大编号
        for(auto& row : hats)
            for(auto& col : row)
                maxhats = max(maxhats,col);
        
        vector<vector<int> > tmp(maxhats + 1);
        //把人对应帽子变成帽子对应人
        for(int i = 0; i < len; i++)
            for(auto& col : hats[i])
                tmp[col].push_back(i);
        
        //全1 1111111111  表示都带着帽子
        //全0 0000000000  表示没有人戴帽子
        vector<vector<int> > dp(maxhats + 1, vector<int>(1<<len));
        dp[0][0] = 1;

        for(int i = 1; i <= maxhats; i++)
            for(int j = 0; j < (1<<len); j++)//0 -->  1<<len  表示了从没有一个人有帽子到全部人都带了帽子的状态
            {
                dp[i][j] = dp[i-1][j];//还没有使用这个帽子，顺延上一次的状态

                for(auto& k : tmp[i])//遍历第i个帽子可以带的人
                    if(j & (1<<k)) //这个人正好带着这个帽子
                        dp[i][j] = (dp[i-1][j ^ (1<<k)] + dp[i][j]) % mod;
                        //累加上前一个帽子时这个人不带帽子的符合的情况，使用异或来一步到位
            }
        
        return dp[maxhats][(1<<len) - 1];
    }
};

修改为一维的dp

这就可以看做是一个01背包的问题

• dp[i]表示当前状态为 i 的所有方案数
• 因为是一维的，所以我们得要从后向前遍历结果，以确保每一次使用到的数据都是前一次dp的数据

	int numberWays(vector<vector<int>>& hats) {
        int len = hats.size();//一共有len个人
        int maxhats = 0;
        //找一个帽子的最大编号
        for(auto& row : hats)
            for(auto& col : row)
                maxhats = max(maxhats,col);
        
        vector<vector<int> > tmp(maxhats + 1);
        //把人对应帽子变成帽子对应人
        for(int i = 0; i < len; i++)
            for(auto& col : hats[i])
                tmp[col].push_back(i);
        
        //dp[i] 表示当前状态为i的所有方案数
        vector<int> dp(1<<len);
        dp[0] = 1;

        for(int i = 1; i <= maxhats; i++)
            for(int j = (1<<len)-1; ~j; j--)//压缩为一维需要从后向前遍历，以确保每一次使用的数据都是上一次的数据
            {
                for(auto& k : tmp[i])//遍历第i个帽子可以带的人
                    if(j & (1<<k)) 
                        dp[j] = (dp[j ^ (1<<k)] + dp[j]) % mod;
            }
        
        return dp[(1<<len) - 1];
    }