leetcode: 349. Intersection of Two Arrays

最新推荐文章于 2024-09-13 16:22:29 发布

霍洛维茨

最新推荐文章于 2024-09-13 16:22:29 发布

阅读量344

点赞数

分类专栏： Leetcode 文章标签： leetcode 算法散列表

本文链接：https://blog.csdn.net/weixin_43065897/article/details/121900609

版权

Leetcode 专栏收录该内容

22 篇文章 0 订阅

订阅专栏

leetcode: 349. Intersection of Two Arrays

原题链接
我的写法：使用unordered_map

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ans;
        unordered_map<int, bool> mp;
        for (int x : nums1) mp[x] = true;
        for (int x : nums2) 
            if (mp[x]) {
                ans.push_back(x);
                mp[x] = false;
            }
        return ans;
    }
};

还有种写法（使用unordered_set）：

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> ans;
        unordered_set<int> nums_set(nums1.begin(), nums1.end());
        for (int x : nums2) 
            if (nums_set.find(x) != nums_set.end())
                ans.insert(x);
        return vector<int> (ans.begin(), ans.end());
    }
};