LeetCode刷题记录——第594题（最长和谐子序列）

题目描述

和谐数组是指一个数组里元素的最大值和最小值之间的差别正好是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