LeetCode 349: 两个数组的交集
题目描述
给定两个数组,编写一个函数来计算它们的交集。
【示例 1】
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
【示例 2】
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
说明:
输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。
解题
关联容器
借助map或者set可以不重复地保存一个数组中的数字,然后遍历另一个数组,与关联容器中保存的数字进行对比,如果相同,那么表明该数字是共有的,将该数字保存在新数组中。如果两个数组有多个相同的共有数字呢?那需要在第二个数组对比过程中,发现某个共有数字之后,将该数字保存到新数组的同时,将该数字从关联容器中删除。
map
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_map<int, int> check_map;
//不重复保存
for (auto &i:nums1){
if (check_map[i]==0)
check_map[i] = 1;
}
vector<int> vals;
for (auto &i : nums2){
if (check_map[i]){
vals.emplace_back(i);
check_map[i] = 0; //将该数字记录删除
}
}
return vals;
}
};
set
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
set<int> check_set(nums1.begin(), nums1.end());
vector<int> vals;
for (auto &i : nums2){
if (check_set.erase(i)) //删除已找到的共有数字,若返回0则代表该数字不存在
vals.emplace_back(i);
}
return vals;
}
};
排序+去重
将两个数组排序并去重,这样两个数组前端都有一部分子序列是排序且不重复的,再利用双指针逐个对比,把相同的数字保存到新数组中。
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
sort(nums1.begin(), nums1.end());
auto iter1 = unique(nums1.begin(), nums1.end());
sort(nums2.begin(), nums2.end());
auto iter2 = unique(nums2.begin(), nums2.end());
vector<int> vals;
for (auto it1=nums1.begin(), it2=nums2.begin(); it1 != iter1 && it2 != iter2;){
if (*it1 < *it2)
++it1;
else if (*it1 > *it2)
++it2;
else{
vals.emplace_back(*it1);
++it1;
++it2;
}
}
return vals;
}
};
排序+二分查找(更新)
在刷二分查找板块的时候也遇到了,能想到的思路是先对两个数组排序,在其中一个数组中使用二分查找方法查找是否包含另一个数组中的元素。