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;
}
};