回溯方法解决每个人戴不同的帽子

总共有 n 个人和 40 种不同的帽子,帽子编号从 1 到 40 。

给你一个整数列表的列表 hats ,其中 hats[i] 是第 i 个人所有喜欢帽子的列表。

请你给每个人安排一顶他喜欢的帽子,确保每个人戴的帽子跟别人都不一样,并返回方案数。

由于答案可能很大,请返回它对 10^9 + 7 取余后的结果。

示例 1:

输入:hats = [[3,4],[4,5],[5]]
输出:1
解释:给定条件下只有一种方法选择帽子。
第一个人选择帽子 3,第二个人选择帽子 4,最后一个人选择帽子 5。
示例 2:

输入:hats = [[3,5,1],[3,5]]
输出:4
解释:总共有 4 种安排帽子的方法:
(3,5),(5,3),(1,3) 和 (1,5)
示例 3:

输入:hats = [[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4]]
输出:24
解释:每个人都可以从编号为 1 到 4 的帽子中选。
(1,2,3,4) 4 个帽子的排列方案数为 24 。
示例 4:

输入:hats = [[1,2,3],[2,3,5,6],[1,3,7,9],[1,8,9],[2,5,7]]
输出:111

提示:

n == hats.length
1 <= n <= 10
1 <= hats[i].length <= 40
1 <= hats[i][j] <= 40
hats[i] 包含一个数字互不相同的整数列表。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-ways-to-wear-different-hats-to-each-other
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution {
public:
    const int MOD = 1e9 + 7;
    int n;
    vector<vector<int>> hats;
    int numberWays(vector<vector<int>>& hats_) {
        hats = hats_;
        n = hats.size();
        
        return dfs(0);
    }
    
    // 已经被使用了的帽子。不能再给接下来的人用
    set<int> used;

    // [0, i) 这些人都已经有了帽子。返回值是给接下来的人选帽子的方案数。
    int dfs(int i) {
        if (i == n) {
            // 已经给所有人选了帽子
            return 1;
        }
        
        int cnt = 0;
        // 给第 i 个人选一顶帽子,不能选已经被用了的帽子。
        for (int hat : hats[i]) {
            if (used.count(hat) == 0) {
                // 可以把 hat 给第 i 个人
                used.insert(hat);
                cnt = (cnt + dfs(i + 1)) % MOD;
                
                // 回溯,把 hat 拿回来
                used.erase(hat);
            }
        }
        return cnt;
    }
};

作者:qodjf
链接:https://leetcode-cn.com/problems/number-of-ways-to-wear-different-hats-to-each-other/solution/zhuang-tai-ya-suo-dp-ru-men-by-qodjf/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

这样会超时,但是思想要学习

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值