给定两个数组,写一个方法来计算它们的交集。
例如:
给定 nums1 = [1, 2, 2, 1]
, nums2 = [2,
2]
, 返回 [2, 2]
.
注意:
- 输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
- 我们可以不考虑输出结果的顺序。
思路:对于这种题型,我的第一个思路就是遍历nums2,看nums1中是否有元素与其相等,若相等,则投入交集数组。
但仔细想想就会发现行不通,因为数组中会可能会有重复元素。
既然会有重复元素,我们不妨延伸该思路,你重复,那我就为你保存你出现的次数。怎么保存?很简单,声明一个数组或者STL的map关联容器来保存nums1中每个值出现的次数。然后再用nums2一一比较,若相等,则与该元素关联的次数减一,减为0,就删除该元素。
代码如下:
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
if (nums1.empty())
return {};
if (nums2.empty())
return {};
vector<int> result;
map<int,int> nums;
for (auto i = nums1.cbegin(); i < nums1.cend(); ++i)
{
++nums[*i];
}
for (int i = 0; i < nums2.size(); ++i)
{
if (nums[nums2[i]])
{
result.push_back(nums2[i]);
if (--nums[nums2[i]] == 0)
nums.erase(nums2[i]);
}
}
return result;
}
};