[LeetCode]349. 两个数组的交集

两个数组的交集

Posted by JinFei on September 23, 2021

题目描述

给定两个数组,编写一个函数来计算它们的交集。

Example 1:

输入:nums1 = [1,2,2,1], nums2 = [2,2] 输出:[2]

Example 2:

输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出:[9,4]

Constraints:

  • 输出结果中的每个元素一定是唯一的。
  • 我们可以不考虑输出结果的顺序。

解题思路

  • 简单题
  • 主要是去重操作,我们将重复的,直接给删除即可。

C++代码

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        unordered_set<int> sets;
        for(auto& i : nums1){
            sets.insert(i);
        }
        for(auto& i : nums2){
            if(sets.count(i)){
                res.push_back(i);
                sets.erase(i);  // 防止重复
            }
        }
        return res;
    }
};