题意
求两个数组的交集
思路
- 用两个unordered_set将讲个数组存进去,然后遍历第二个set内的每一个元素,去查看第一个set里面有没有,在存入set的过程中已经去重。
- 先排序,用两个指针i和j分别指向两个数组的开头,并且:
- s1[i]<s2[j]:i++
- s1[i]>s2[j]:j++
- s1[i]==s2[j]:查看结果最后一个元素是否为s1[i](因此已经排过序了,所以只需要查看最后一个元素)。
时间复杂度:O(logn)
代码
//algorithm1: based on hash
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int> s1, s2;
vector<int> res;
for (auto x : nums1) s1.insert(x);
for (auto x : nums2) s2.insert(x);
for (auto x : s2) {
if (s1.find(x) != s1.end()) res.push_back(x);
}
return res;
}
};
//algorithm2: based on sort and two pointers
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
sort(nums1.begin(), nums1.end());
sort(nums2.begin(), nums2.end());
vector<int> res;
int i = 0, j = 0;
while (i < nums1.size() && j < nums2.size()) {
if (nums1[i] == nums2[j]) {
if (res.empty() || res[res.size() - 1] != nums1[i]) res.push_back(nums1[i]);
i++; j++;
}
else if (nums1[i] < nums2[j]) i++;
else j++;
}
return res;
}
};