LeetCode 1681. 最小不兼容性(回溯+剪枝)
生活随笔
收集整理的這篇文章主要介紹了
LeetCode 1681. 最小不兼容性(回溯+剪枝)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
文章目錄
- 1. 題目
- 2. 解題
1. 題目
給你一個整數數組 nums??? 和一個整數 k 。你需要將這個數組劃分到 k 個相同大小的子集中,使得同一個子集里面沒有兩個相同的元素。
一個子集的 不兼容性 是該子集里面最大值和最小值的差。
請你返回將數組分成 k 個子集后,各子集 不兼容性 的 和 的 最小值 ,如果無法分成分成 k 個子集,返回 -1 。
子集的定義是數組中一些數字的集合,對數字順序沒有要求。
示例 1: 輸入:nums = [1,2,1,4], k = 2 輸出:4 解釋:最優的分配是 [1,2] 和 [1,4] 。 不兼容性和為 (2-1) + (4-1) = 4 。 注意到 [1,1] 和 [2,4] 可以得到更小的和,但是第一個集合有 2 個相同的元素,所以不可行。示例 2: 輸入:nums = [6,3,8,1,3,1,2,2], k = 4 輸出:6 解釋:最優的子集分配為 [1,2],[2,3],[6,8] 和 [1,3] 。 不兼容性和為 (2-1) + (3-2) + (8-6) + (3-1) = 6 。示例 3: 輸入:nums = [5,3,3,6,3,3], k = 3 輸出:-1 解釋:沒辦法將這些數字分配到 3 個子集且滿足每個子集里沒有相同數字。提示: 1 <= k <= nums.length <= 16 nums.length 能被 k 整除。 1 <= nums[i] <= nums.length來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/minimum-incompatibility
著作權歸領扣網絡所有。商業轉載請聯系官方授權,非商業轉載請注明出處。
2. 解題
寫的回溯,超時了,參考評論區的劉波同學的代碼,做了剪枝,通過了
class Solution {int mindiff = INT_MAX;int size;//每個集合的大小 public:int minimumIncompatibility(vector<int>& nums, int k) {if(k == nums.size())//特殊情況return 0;unordered_map<int, int> m;sort(nums.begin(), nums.end());for(auto n : nums){if(++m[n] > k)return -1;}size = nums.size()/k;vector<int> curnum(k), count(k, 0);//每個集合當前的數,集合大小dfs(nums, curnum, count, 0, 0);return mindiff;} void dfs(vector<int>& nums, vector<int>& curnum, vector<int>& count, int i, int diff){if(diff >= mindiff)//剪枝return;if(i == nums.size()){if(diff < mindiff)mindiff = diff;return;}for(int j = 0; j < count.size(); ++j){if(curnum[j] == nums[i])//存在相同的數字了continue;if(count[j] < size)//集合還沒滿{int originnum = curnum[j];//集合當前的數字curnum[j] = nums[i];//nums[i] 放入集合 jcount[j]++;//集合大小 + 1int delta = count[j] >= 2 ? (nums[i]-originnum) : 0;// diff 增量dfs(nums, curnum, count, i+1, diff+delta);count[j]--;curnum[j] = originnum;if(count[j] == 0)//參考 劉波 同學//都是空的集合,這個數字放在哪個集合都是一樣的,不寫這句,會超時break;}}} };// 28 ms 7.6 MB C++28 ms 7.6 MB C++
我的CSDN博客地址 https://michael.blog.csdn.net/
長按或掃碼關注我的公眾號(Michael阿明),一起加油、一起學習進步!
總結
以上是生活随笔為你收集整理的LeetCode 1681. 最小不兼容性(回溯+剪枝)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: LeetCode 967. 连续差相同的
- 下一篇: LeetCode 729. 我的日程安排