LeetCode 1681. 最小不兼容性(回溯+剪枝)
文章目錄
- 1. 題目
- 2. 解題
1. 題目
給你一個(gè)整數(shù)數(shù)組 nums??? 和一個(gè)整數(shù) k 。你需要將這個(gè)數(shù)組劃分到 k 個(gè)相同大小的子集中,使得同一個(gè)子集里面沒(méi)有兩個(gè)相同的元素。
一個(gè)子集的 不兼容性 是該子集里面最大值和最小值的差。
請(qǐng)你返回將數(shù)組分成 k 個(gè)子集后,各子集 不兼容性 的 和 的 最小值 ,如果無(wú)法分成分成 k 個(gè)子集,返回 -1 。
子集的定義是數(shù)組中一些數(shù)字的集合,對(duì)數(shù)字順序沒(méi)有要求。
示例 1: 輸入:nums = [1,2,1,4], k = 2 輸出:4 解釋:最優(yōu)的分配是 [1,2] 和 [1,4] 。 不兼容性和為 (2-1) + (4-1) = 4 。 注意到 [1,1] 和 [2,4] 可以得到更小的和,但是第一個(gè)集合有 2 個(gè)相同的元素,所以不可行。示例 2: 輸入:nums = [6,3,8,1,3,1,2,2], k = 4 輸出:6 解釋:最優(yōu)的子集分配為 [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 解釋:沒(méi)辦法將這些數(shù)字分配到 3 個(gè)子集且滿(mǎn)足每個(gè)子集里沒(méi)有相同數(shù)字。提示: 1 <= k <= nums.length <= 16 nums.length 能被 k 整除。 1 <= nums[i] <= nums.length來(lái)源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/minimum-incompatibility
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有。商業(yè)轉(zhuǎn)載請(qǐng)聯(lián)系官方授權(quán),非商業(yè)轉(zhuǎn)載請(qǐng)注明出處。
2. 解題
寫(xiě)的回溯,超時(shí)了,參考評(píng)論區(qū)的劉波同學(xué)的代碼,做了剪枝,通過(guò)了
class Solution {int mindiff = INT_MAX;int size;//每個(gè)集合的大小 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);//每個(gè)集合當(dāng)前的數(shù),集合大小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])//存在相同的數(shù)字了continue;if(count[j] < size)//集合還沒(méi)滿(mǎn){int originnum = curnum[j];//集合當(dāng)前的數(shù)字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)//參考 劉波 同學(xué)//都是空的集合,這個(gè)數(shù)字放在哪個(gè)集合都是一樣的,不寫(xiě)這句,會(huì)超時(shí)break;}}} };// 28 ms 7.6 MB C++28 ms 7.6 MB C++
我的CSDN博客地址 https://michael.blog.csdn.net/
長(zhǎng)按或掃碼關(guān)注我的公眾號(hào)(Michael阿明),一起加油、一起學(xué)習(xí)進(jìn)步!
總結(jié)
以上是生活随笔為你收集整理的LeetCode 1681. 最小不兼容性(回溯+剪枝)的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: LeetCode 967. 连续差相同的
- 下一篇: LeetCode 729. 我的日程安排