LeetCode刷题-数组594：最长和谐子序列

题目

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

示例3：

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

提示：

• 1 <= nums.length <= 2 * 104
• -109 <= nums[i] <= 109

解题思路

法一：哈希表（执行用时：336ms，内存消耗：15.3MB）

        先利用collection.Counter(nums)将数组中各个数字及其个数以键对形式存入哈希表中，用res保存和谐子序列的长度，然后遍历哈希表，如果比当前值大1的值也在哈希表中，则更新和谐子序列的长度。

class Solution(object):
    def findLHS(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        res = 0
        hashtable = collections.Counter(nums)
        for i in hashtable:
            if i+1 in hashtable:
                res = max(res, hashtable[i]+hashtable[i+1])
        return res

 

时间复杂度：O(N)，其中 N 为数组的长度。

空间复杂度：O(N)，其中 N 为数组的长度。数组中最多有 N 个不同元素，因此哈希表最多存储 N 个数据。

法二：排序双指针（执行用时：360ms，内存消耗：13.9MB）

        因为题目只要求返回最长和谐序列的长度，所以可以先将数组按从小到大进行排序，然后依次找到相邻两个连续相同元素的子序列，并检查这两个子序列的元素之差是否为1。

        有点类似滑动窗口的做法，将begin指向第一个连续相同元素的子序列的第一个元素，end指向相邻的第二个相同元素的子序列的末尾元素，如果满足二者的元素之差为1，则当前的和谐子序列的长度即为两个子序列的长度之和，即end - begin +1，最后返回最大值即可。

class Solution(object):
    def findLHS(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        nums.sort()
        res,begin = 0,0
        for end in range(1,len(nums)):
            #右指针不断移动，一旦发现右指针数据不满足差值为1这个条件了，左指针就前进，直到满足要求
            while nums[end] - nums[begin] > 1:
                begin += 1
            #当这一段符合要求的最长和谐子序列已经找到，就记录这一段的长度，以此不断更新
            if nums[end] - nums[begin] == 1:
                res = max(res, end-begin+1)
        return res

时间复杂度：O(Nlog⁡N)，其中 N 为数组的长度。我们首先需要对数组进行排序，花费的时间复杂度为 O(Nlog⁡N)，我们需要利用双指针遍历数组花费的时间为 O(2N)，总的时间复杂度 T(N)=O(Nlog⁡N)+O(2N)=O(Nlog⁡N)。

空间复杂度：O(1)，需要常数个空间保存中间变量。