最长和谐子序列
题目描述:
和谐数组是指一个数组里元素的最大值和最小值之间的差别 正好是 1 。
现在,给你一个整数数组 nums ,请你在所有可能的子序列中找到最长的和谐子序列的长度。
数组的子序列是一个由数组派生出来的序列,它可以通过删除一些元素或不删除元素、且不改变其余元素的顺序而得到。
示例 :
输入:nums = [1,3,2,2,5,2,3,7]
输出:5
解释:最长的和谐子序列是 [3,2,2,2,3]
提示:
- 1 <= nums.length <= 2 * 104
- -109 <= nums[i] <= 109
解法
可以使用 hash 表来记录各数字的次数,然后找相邻两个数字次数和的最大值即可 。注意,差值必须是 1,而不是小于等于 1。
代码
class Solution:
def findLHS(self, nums: List[int]) -> int:
num2cnt = Counter(nums)
res = 0
for num in num2cnt.keys():
if num + 1 in num2cnt:
res = max(res, num2cnt[num] + num2cnt[num + 1])
return res
测试结果
执行用时:260 ms, 在所有 Python3 提交中击败了 90.70% 的用户
内存消耗:16.5 MB, 在所有 Python3 提交中击败了 56.20% 的用户
说明
算法题来源:力扣(LeetCode)