1413. 逐步求和得到正数的最小值
给你一个正整数数组 nums,请你帮忙从该数组中找出能满足下面要求的 最长 前缀,并返回该前缀的长度:
- 从前缀中 恰好删除一个 元素后,剩下每个数字的出现次数都相同。 如果删除这个元素后没有剩余元素存在,仍可认为每个数字都具有相同的出现次数(也就是 0 次)。
示例1
输入:nums = [2,2,1,1,5,3,3,5]
输出:7
解释:对于长度为 7 的子数组 [2,2,1,1,5,3,3],如果我们从中删去 nums[4] = 5,就可以得到 [2,2,1,1,3,3],里面每个数字都出现了两次。
示例2
输入: nums = [1,1,1,2,2,2,3,3,3,4,4,4,5]
输出: 13
思路
解法
func maxEqualFreq(nums []int) int {
dic:=map[int]int{}
freq:=map[int]int{}
res:=0
maxn:=0
for i:=0;i<len(nums);i++{
dic[nums[i]]++
maxn=max(maxn,dic[nums[i]])
freq[dic[nums[i]]]++
if freq[maxn]==1 && freq[maxn-1]*(maxn-1)==i{
res=i+1
}else if freq[maxn]*maxn==i{
res=i+1
}
}
if maxn==1{
return len(nums)
}else{
return res
}
}
func max(a,b int)int{
if a>b{
return a
}
return b
}