LeetCode 594 最长和谐子序列[双指针Map] HERODING的LeetCode之路

Posted 2021-12-15 HERODING23

解题思路：
既然可以删除元素，那么所谓的不改变顺序这个条件就是幌子，直接sort排序，用双指针的思想，或者说滑动窗口的思想，begin固定最小值，i不断往后遍历，作为最大值的下标，一旦不满足条件，begin++，满足条件更新最大长度，代码如下：

class Solution 
public:
    int findLHS(vector<int>& nums) 
        sort(nums.begin(), nums.end());
        int begin = 0;
        int res = 0;
        for(int i = 0; i < nums.size(); i ++) 
            while(nums[i] - nums[begin] > 1) 
                begin ++;
            
            if(nums[i] - nums[begin] == 1) 
                res = max(res, i - begin + 1);
            
        
        return res;
    
;

时间复杂度：O(NlogN)
空间复杂度：O(1)

当然还有更简单的实现，利用map，直接实现映射，观察前后两个差为1的数的个数之和，找到最大的返回即可，代码如下：

class Solution 
public:
    int findLHS(vector<int>& nums) 
        unordered_map<int, int> mp;
        int res = 0;
        for(auto& num : nums) 
            mp[num] ++;
        
        for(auto& [a, b] : mp) 
            if(mp.count(a + 1)) 
                res = max(res, b + mp[a + 1]);
            
        
        return res;
    
;

时间复杂度：O(N)
空间复杂度：O(N)