力扣第 158 场周赛第 4 题
题目
给你一个正整数数组 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
提示:
2 <= nums.length <= 105
1 <= nums[i] <= 105
相似问题:
分析
#1
- 维护每个数的次数,设当前的次数数组为 A
- 删除操作等价于从 A 选一个元素减 1
- 统计 A 的元素的频数,分类讨论
- 假如元素种类>2,不可能满足
- 假如元素种类=2,分别是 a,b
- 要么 a=1 且 a 的频数为 1
- 要么 a=b-1 且 b 的频数为 1
- 假如元素种类=1,为 a
- 因此维护 A 的计数器即可(二重计数)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
class Solution:
def maxEqualFreq(self, nums: List[int]) -> int:
def check(cct):
if len(cct)>2:
return False
if len(cct)==1:
a,w = list(cct.items())[0]
return a==1 or w==1
a,b = sorted(cct)
return a==cct[a]==1 or (a==b-1 and cct[b]==1)
ct = defaultdict(int)
cct = defaultdict(int)
res = 0
for i,x in enumerate(nums):
a = ct[x]
ct[x] += 1
if a>0:
cct[a] -= 1
if not cct[a]:
del cct[a]
cct[a+1] += 1
if check(cct):
res = i+1
return res
|
199 ms
#2
由于只求最长,可以倒序遍历到第一个符合的即可返回
解答
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
class Solution:
def maxEqualFreq(self, nums: List[int]) -> int:
def check(cct):
if len(cct)>2:
return False
if len(cct)==1:
a,w = list(cct.items())[0]
return a==1 or w==1
a,b = sorted(cct)
return a==cct[a]==1 or (a==b-1 and cct[b]==1)
n = len(nums)
ct = Counter(nums)
cct = Counter(ct.values())
for i in range(n-1,-1,-1):
if check(cct):
return i+1
x = nums[i]
a = ct[x]
ct[x] -= 1
cct[a] -= 1
if not cct[a]:
del cct[a]
if a-1>0:
cct[a-1] += 1
return 0
|
32 ms