LeetCode 1655. 分配重复整数(回溯)

1. 题目

给你一个长度为 n 的整数数组 nums ,这个数组中至多有 50 个不同的值。
同时你有 m 个顾客的订单 quantity ,其中,整数 quantity[i] 是第 i 位顾客订单的数目。请你判断是否能将 nums 中的整数分配给这些顾客,且满足:

  • 第 i 位顾客 恰好 有 quantity[i] 个整数。
  • 第 i 位顾客拿到的整数都是 相同的 。
  • 每位顾客都满足上述两个要求。

如果你可以分配 nums 中的整数满足上面的要求,那么请返回 true ,否则返回 false 。

示例 1:
输入:nums = [1,2,3,4], quantity = [2]
输出:false
解释:第 0 位顾客没办法得到两个相同的整数。

示例 2:
输入:nums = [1,2,3,3], quantity = [2]
输出:true
解释:第 0 位顾客得到 [3,3] 。整数 [1,2] 都没有被使用。

示例 3:
输入:nums = [1,1,2,2], quantity = [2,2]
输出:true
解释:第 0 位顾客得到 [1,1] ,第 1 位顾客得到 [2,2] 。

示例 4:
输入:nums = [1,1,2,3], quantity = [2,2]
输出:false
解释:尽管第 0 位顾客可以得到 [1,1] ,第 1 位顾客没法得到 2 个一样的整数。

示例 5:
输入:nums = [1,1,1,1,1], quantity = [2,3]
输出:true
解释:第 0 位顾客得到 [1,1] ,第 1 位顾客得到 [1,1,1] 。
 
提示:
n == nums.length
1 <= n <= 10^5
1 <= nums[i] <= 1000
m == quantity.length
1 <= m <= 10
1 <= quantity[i] <= 10^5
nums 中至多有 50 个不同的数字。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/distribute-repeating-integers
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

2.1 回溯

class Solution {
    bool ans = false;
public:
    bool canDistribute(vector<int>& nums, vector<int>& quantity) {
        unordered_map<int,int> m;
        for(auto n : nums)
            m[n]++;
        vector<int> val(m.size(), 0);
        int i = 0;
        for(auto it = m.begin(); it != m.end(); ++it)
            val[i++] = it->second;
        sort(val.rbegin(), val.rend());
        sort(quantity.rbegin(), quantity.rend());
        dfs(val, quantity, 0);
        return ans;
    }
    void dfs(vector<int>& val, vector<int>& quantity, int i)
    {
        if(ans)
            return;
        if(i == quantity.size())
        {
            ans = true;
            return;
        }
        for(int j = 0; j < val.size(); ++j)
        {
            if(val[j]-quantity[i] >= 0)
            {
                val[j] -= quantity[i];
                dfs(val, quantity, i+1);
                val[j] += quantity[i];
            }
        }
    }
};

820 ms 73.5 MB


我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!
Michael阿明

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页