【LeetCode】594. 最长和谐子序列

最新推荐文章于 2024-01-22 23:12:50 发布

原创最新推荐文章于 2024-01-22 23:12:50 发布 · 214 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种利用C++ map迭代器自动排序特性，来解决寻找最长和谐子序列问题的方法。通过统计数组中各元素出现的频率，再按升序遍历map，找到满足条件的相邻元素对，从而得到最长和谐子序列的长度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

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

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

示例 1:

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

说明: 输入的数组长度最大不超过20,000.

思路

利用map的迭代器自动排序的特点，先统计每个数字出现的次数，再从小到大求一个最大值。

代码

class Solution {
public:
    int findLHS(vector<int>& nums) {
        if(nums.empty()) return 0;
        int res = 0;
        int len = nums.size();
        map<int, int> m;
        for (int i = 0; i < len; i++)
            m[nums[i]]++;
        map<int, int>::iterator it = m.begin();
        it++;
        for (; it != m.end(); ++it) {
            auto pre = prev(it);    
            //next(it)返回下一个迭代器指针，prev(it)，返回上一个迭代器指针
            if (it->first == pre->first + 1) {
                res = max(res, it->second + pre->second);
            }
        }
        return res;
    }
};