python笔记(2) --- LeetCode 594. 最长和谐子序列

题目来自力扣594. 最长和谐子序列 - 力扣（LeetCode）

记录每日刷题学习过程，如有侵权问题，联系作者删除。  

欢迎有意向的同学们交流学习。

一、题目描述

和谐数组是指一个数组里元素的最大值和最小值之间的差别 正好是 1 。

给你一个整数数组 nums ，请你在所有可能的 子序列 中找到最长的和谐子序列的长度。

数组的 子序列 是一个由数组派生出来的序列，它可以通过删除一些元素或不删除元素、且不改变其余元素的顺序而得到。

示例1：

输入：nums = [1,3,2,2,5,2,3,7]

输出：5

解释：

最长和谐子序列是 [3,2,2,2,3]。

示例2：

输入：nums = [1,2,3,4]

输出：2

解释：

最长和谐子序列是 [1,2]，[2,3] 和 [3,4]，长度都为 2。

示例3：

输入：nums = [1,1,1,1]

输出：0

解释：

不存在和谐子序列。

二、思路笔记

题目要求的和谐子序列中，最大值和最小值的差为1，求得给定数组中的最长和谐子序列。

1. 遍历原始数组中的值，构造一个字典，key为可能的元素，value为元素在数组中出现的次数。

2. 对于每个不重复的元素，如果与它差值的绝对值为1的数字同时出现在数组中，则将其出现的次数求和即为子序列的长度，每遍历一个元素则刷新一次子序列长度的值。

三、代码

class Solution:
    def findLHS(self, nums: List[int]) -> int:
        nums_set = set(nums)
        mx = 0
        for j in nums_set:
            if j - 1 in nums_set:
                if nums.count(j) + nums.count(j - 1) > mx:
                    mx = nums.count(j) + nums.count(j - 1)
            elif j + 1 in nums_set:
                if nums.count(j) + nums.count(j + 1) > mx:
                    mx = nums.count(j) + nums.count(j + 1)
        return mx