题目链接:https://leetcode-cn.com/problems/longest-harmonious-subsequence/
题目分析:
题目中说可以通过删除一些元素或不删除元素、且不改变其余元素的顺序而得到。
说到不改变其他元素的顺序,但是答案最终返回的是数组的长度,并且可以删除或者不删除其中的一些元素,那么就可以无视顺序进行操作。
法一(伪滑动窗口):
首先对数组进行排序。
设置两个指针维护一个窗口。
由于已经在初始时对数组进行排序,所以当前窗口的最左端元素一定是最小值,同理最右端的元素一定是最大值。
设置左指针 left 和右指针right。
-
如果当前right所指的值 减去 left所指的值 =1 则更新当前窗口内的最大长度值。然后继续扩大窗口。right+=1
-
如果right所指的值 减去 left所指的值 = 0 则扩大窗口内的元素 right+=1。
注意:这里并不更新最大长度的值,预防【1,1,1,1】的情况。
-
right所指的值 减去 left所指的值 >1 则left+=1 左指针往后挪。
完整代码:
class Solution:
def findLHS(self, nums: List[int]) -> int:
nums.sort()
# 滑动窗口
i = 0
j = 1
res = 0
temp = False
while j < len(nums):
if nums[j] - nums[i] == 1 or nums[j] - nums[i] == 0:
if nums[j] - nums[i] == 1:
res = max(res,j-i+1)
j +=1
else:
i +=1
print(res)
return res
法二(哈希表):
答案只需要我们返回数组长度,即数组中有多少个数。
使用哈希表 key存数值 对应value存该数值出现的次数,
并且让哈希表根据key进行排序。
这样我们只需要算出来相邻两个key的差值是否为1,如果为1则算出来他们两个value的总和,并不断更新总和的最大值即可。
完整代码
class Solution:
def findLHS(self, nums: List[int]) -> int:
from collections import Counter
res = 0
hs = Counter(nums)
# 哈希表根据key进行排序, 后面的匿名函数换成 x:x【1】就是根据value排序了.
hs = sorted(hs.items(),key= lambda x:x[0])
print(hs)
for i in range(len(hs)-1):
if hs[i+1][0] - hs[i][0] == 1:
res = max(res,hs[i][1]+hs[i+1][1])
print(res)
return res