力扣:349. 两个数组的交集
题目:
给定两个数组 nums1 和 nums2 ,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
思路:
将其中一个数组读入到一个set容器中,然后依次取另一个数组中的每一个元素看是否在此set容器中,若在则放入set2容器中,最后将set2容器中的元素放入vector容器中且返回。
set容器的选择:
std::set和std::multiset底层实现都是红黑树,std::unordered_set的底层实现是哈希表,使用unordered_set 读写效率是最高的,并不需要对数据进行排序,而且还不要让数据重复,所以选择unordered_set。
代码:
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int>s(nums1.begin(),nums1.end());
unordered_set<int>result_set;
for(int ret:nums2){
if(s.find(ret) != s.end()){
result_set.insert(ret);
}
}
return vector<int>(result_set.begin(),result_set.end());
}
};