[LeetCode]594. Longest Harmonious Subsequence

最新推荐文章于 2021-12-13 11:09:55 发布

原创最新推荐文章于 2021-12-13 11:09:55 发布 · 793 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 专栏收录该内容

184 篇文章

订阅专栏

本文介绍了解决LeetCode上第594题“最长和谐子序列”的方法，通过使用哈希表统计元素出现次数，实现快速查找符合和谐条件的数对，给出了一段C++代码示例。

[LeetCode]594. Longest Harmonious Subsequence

题目描述

这里写图片描述

思路

哈希表统计每个元素出现的次数

代码

#include <iostream>
#include <unordered_map>
#include <vector>
#include <algorithm>

using namespace std;

class Solution {
public:
    int findLHS(vector<int>& nums) {
        unordered_map<int, int> um;
        for (int num : nums)
            ++um[num];
        int res = 0;
        for (auto p : um) {
            if (um.count(p.first - 1))
                res = max(res, p.second + um[p.first - 1]);
            if (um.count(p.first + 1))
                res = max(res, p.second + um[p.first + 1]);
        }
        return res;
    }
};

int main() {
    vector<int> nums = { 1,1,1,1 };
    Solution s;
    cout << s.findLHS(nums) << endl;

    system("pause");
    return 0;
}