程式語言 - LeetCode - C++ - 594. Longest Harmonious Subsequence



題目:


解答:

class Solution {
public:
    int findLHS(vector<int>& nums) {
        unordered_map<int, int> mp;

        for (int n : nums) {
            mp[n] += 1;
        }

        int ans = 0;

        for (auto& [k, v] : mp) {
            if (mp.count(k + 1)) {
                ans = max(ans, v + mp[k + 1]);
            }
        }

        return ans;
    }
};