每日力扣-哈希-最长和谐子序列

76 阅读1分钟

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

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

输入: [1,3,2,2,5,2,3,7] 输出: 5 原因: 最长的和谐数组是:[3,2,2,2,3].

解题思路:先放进map,然后逐一比较

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