给定两个数组,写一个函数来计算它们的交集。
例子:
给定 num1= [1, 2, 2, 1]
, nums2 = [2, 2]
, 返回 [2]
.
提示:
- 每个在结果中的元素必定是唯一的。
- 我们可以不考虑输出结果的顺序。
思路:
这道题用set就可以做,时间复杂度是O(n(nums1)+n(nums2)),不知道为什么放在Binary Search中,一般二分查找中都要求数组排序,而排序一般情况最低都是O(nlogn)的时间复杂度。所以这道题用set来做即可,简单高效。
把数组nums1的元素放到一个set中,然后遍历nums2数组,如果在set中存在就放到res中,最后返回res。
代码如下:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
set<int> s(nums1.begin(),nums1.end()),res_tmp;
for (auto m : nums2) {
if (s.count(m)) {
res_tmp.insert(m);
}
}
vector<int> res(res_tmp.begin(), res_tmp.end());
return res;
}