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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果

代码描述

思路：把原数组，放入map中，顺序扫描查找相差为1的两个键，同时记录两个键的数量之和，是否为最大。最后输出最大值即可。

class Solution {
public:
    int findLHS(vector<int>& nums) {
        if(nums.size() < 2) return 0;
        map<int, int> serice;
        for(int i = 0; i < nums.size(); ++i)
        {
            serice[nums[i]]++;
        }
        map<int, int>::iterator it1 = serice.begin();
        map<int, int>::iterator it2 = serice.begin();
        ++it2;
        int res = 0;
        for(; it2 != serice.end(); ++it2, ++it1)
        {
            if(abs(it1->first - it2->first) == 1)
            {
                res < (it1->second + it2->second) ? res = (it1->second + it2->second) : res;
            }
        }
        return res;
    }
};