题目描述:
给你一个长度为 n
下标从 0 开始的整数数组 nums
。
我们想将下标进行分组,使得 [0, n - 1]
内所有下标 i
都 恰好 被分到其中一组。
如果以下条件成立,我们说这个分组方案是合法的:
- 对于每个组
g
,同一组内所有下标在nums
中对应的数值都相等。 - 对于任意两个组
g1
和g2
,两个组中 下标数量 的 差值不超过1
。
请你返回一个整数,表示得到一个合法分组方案的 最少 组数。
示例 1:
输入:nums = [3,2,3,2,3]
输出:2
解释:一个得到 2 个分组的方案如下,中括号内的数字都是下标:
组 1 -> [0,2,4]
组 2 -> [1,3]
所有下标都只属于一个组。
组 1 中,nums[0] == nums[2] == nums[4] ,所有下标对应的数值都相等。
组 2 中,nums[1] == nums[3] ,所有下标对应的数值都相等。
组 1 中下标数目为 3 ,组 2 中下标数目为 2 。
两者之差不超过 1 。
无法得到一个小于 2 组的答案,因为如果只有 1 组,组内所有下标对应的数值都要相等。
所以答案为 2 。
示例 2:
输入:nums = [10,10,10,3,1,1]
输出:4
解释:一个得到 2 个分组的方案如下,中括号内的数字都是下标:
组 1 -> [0]
组 2 -> [1,2]
组 3 -> [3]
组 4 -> [4,5]
分组方案满足题目要求的两个条件。
无法得到一个小于 4 组的答案。
所以答案为 4 。
提示:
1 <= nums.length <= 105
1 <= nums[i] <= 109
思路:
题目要求我们求出最小的分组数目,首先我们可以确定的是,一个数组一定有答案,因为再不济把他每个元素都分一个组就可以了,又由于对于任意两个组 g1
和 g2
,两个组中 下标数量 的 差值不超过1,假设我们分的所有的组的下标数量最小为k,那么最大也只能是k+1.
我们可以先用一个map存下每一个数出现的次数,对于每一个数it,出现的次数为cnt,我们看能否将这cnt个it分为只包含k个和k+1个的组。
假设数组中出现的最小次数的数目为mi,那么很容易得到k<=mi.如果k>mi,意味着至少有一组凑不够k个相同的数(同一组内所有下标在 nums 中对应的数值),所以我们可以遍历k(分的所有的组的下标数量最小值),k确定了,k+1也就确定了。
对于每一个k,我们看nums里的所有数出现的次数能不能分为只包含k个和k+1个的组。
如果可以,我们就把当前k可以分得的最小组的数目求一个最小值ans.
如果不能那就。。。那就不能。
时间复杂读为什么是O(n)?
假设t表示的是nums数组中不同元素的个数,那么最小出现次数mi<=n/t,所以mi*t<=n.
O(min(mp[nums[i])*t)=O(mi*t)=O(n/t *t)=O(n)
这里计算时间复杂度非常重要哦,我开始也是算错了时间复杂度以为是o(n^2)了。、
代码:
class Solution { public: int minGroupsForValidAssignment(vector<int>& nums) { map<int,int> mp; int len=nums.size(); int mi=1e9+10;//最少出现次数 for(auto it:nums){ mp[it]++;//记录每个元素出现的次数 } for(auto it:mp){ mi=min(it.second,mi); } int ans=1e9; for(int k=1;k<=mi;k++){ int tn=0;//记录当前k可以分得到的最小数目的组数 int f=1; for(auto it:mp){ int cnt=it.second; int a = cnt / (k + 1); int b = cnt - a * (k + 1); if (cnt % (k + 1) == 0) {//尽量拼元素多的组 tn += a; } else if (a + b >= k){//看能不能在a个(k+1)的组里面分诺干个1到剩下的b里面 tn += a + 1; }else{//优先拼k+1组失败了,退而求其次优先考虑拼k组 int a = cnt / (k ); int b = cnt - a * (k); if(b>a){//如果剩下的元素b不能分到a个(k)组里面,说明分组失败 f=0; break; } tn+=a; } } if(f) ans=min(ans,tn); } return ans; } };