Leetcode 594：最长和谐子序列

题目描述

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

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

示例 1:

输入: [1,3,2,2,5,2,3,7]
输出: 5
原因: 最长的和谐数组是：[3,2,2,2,3].
说明: 输入的数组长度最大不超过20,000.

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/longest-harmonious-subsequence
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

解题思路

解法一：

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

解法二：

class Solution {
public:
    int findLHS(vector<int>& nums) {
        if(nums.size() == 0) return 0;
        map<int,int> mp;
        int ans = 0,len = nums.size();
        for(int i=0;i<len;i++) mp[nums[i]]++;
        auto itpre = mp.begin();
        auto itpost = mp.begin();
        itpost++;
        while(itpost != mp.end()){
            if(itpre->first+1 == itpost->first) ans = max(ans,itpre->second+itpost->second);
            itpre++;
            itpost++;
        }
        return ans;
    }
};