Given two arrays, write a function to compute their intersection.
分析:
本题与349题较为相似,唯一不同之处在于交集允许出现重复数字。因此,我们不仅需要记录集合中有哪些数字出现,也需要记录数字出现的频次。这里可以采用map来实现,map的键值可以记录哪些数字出现,键值对应数字记录频次。
代码:
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
map<int, int> record;
for (int i = 0; i < nums1.size(); i++)
{
record[nums1[i]]++;
}
vector<int> resultVector;
for (int i = 0; i < nums2.size(); i++)
{
if (record[nums2[i]] > 0 )
{
resultVector.push_back(nums2[i]);
record[nums2[i]]--;
}
}
return resultVector;
}
};