程式語言 - LeetCode - CPP - 350. Intersection of Two Arrays II



題目:


解答:

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ans;
        unordered_map<int, int> cnt;

        for (int n : nums1) {
            cnt[n] += 1;
        }

        for (int n : nums2) {
            if (cnt[n] > 0) {
                ans.push_back(n);
                cnt[n] -= 1;
            }
        }

        return ans;
    }
};