题目描述
和谐数组是指一个数组里元素的最大值和最小值之间的差别正好是1。
现在,给定一个整数数组,你需要在所有可能的子序列中找到最长的和谐子序列的长度。
示例 1:
输入: [1,3,2,2,5,2,3,7]
输出: 5
原因: 最长的和谐数组是:[3,2,2,2,3].
思路分析
- 利用collections.Counter,将列表统计建立字典,key为列表中的值,value为相应的值出现的次数
- 遍历列表,对于列表中的值num,如果num+1在列表中,并且出现的次数不为0,那么我们就需要考虑它是不是最大的和谐子序列
- 用res记录最大子序列的长度,由于规定序列最大最小值的差为1,所以d[num] + d[num+1] 正好满足了要求
代码示例
class Solution(object):
def findLHS(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
res = 0
d = collections.Counter(nums)
for num in nums:
if num + 1 in d:
res = max(res, d[num] + d[num + 1])
return res