LeetCode 349. Intersection of Two Arrays 题解（C++）

LeetCode 349. Intersection of Two Arrays 题解（C++）

---

题目描述

• Given two arrays, write a function to compute their intersection.

示例

• Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2].

补充

• Each element in the result must be unique.
• The result can be in any order.

思路

• 使用stl的unordered_set实现。unordered_set是一个存储唯一元素的关联元素，且使用hash表来组织它们的元素，使得查找效率为O（1）；
• 先创建一个unordered_set容器num，并将nums1的元素放入其中，之后再对nums2进行遍历，若在容器num中找到该元素，则删除容器中的元素，并将该元素放入存放结果的数组中。

代码

class Solution 
{
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) 
    {
        unordered_set<int> num;
        vector<int> result;
        for (int i = 0; i < nums1.size(); ++i)
        {
            num.insert(nums1[i]);   
        }
        for (int i = 0; i < nums2.size(); ++i)
        {
            unordered_set<int>::iterator it = num.find(nums2[i]);
            if (it != num.end())
            {
                num.erase(nums2[i]);
                result.push_back(nums2[i]);
            }
        }
        return result;
    }
};