leetcode第349题. 两个数组的交集
题意:给定两个数组,编写一个函数来计算它们的交集。
思路
注意说明中给出:输出结果中的每个元素一定是唯一的。说明也就是说输出的结果的去重的, 同时可以不考虑输出结果的顺序。
可以发现,貌似用数组做哈希表可以解决这道题目(像昨天那样),把nums1的元素,映射到哈希数组的下标上,然后在遍历nums2的时候,判断是否出现过就可以了。
但是要注意,使用数组来做哈希的题目,都限制了数值的大小。昨天的题目中只有小写字母,或者数值大小在[0- 10000] 之内等等。
而这道题目没有限制数值的大小,就无法使用数组来做哈希表了。
而且如果哈希值比较少、特别分散、跨度非常大,使用数组就造成空间的极大浪费。
此时就要使用另一种结构体了,set ,关于set,C++ 给提供了如下三种可用的数据结构:
① std::set
② std::multiset
③ std::unordered_set
std::set和std::multiset底层实现都是红黑树,std::unordered_set的底层实现是哈希表, 使用unordered_set 读写效率是最高的,并不需要对数据进行排序,而且还不要让数据重复,所以选择unordered_set。
c++代码
class solution
{
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2)
{
unordered_set<int> nums(nums1.begin(), nums1.end()); //通过nums1数组初始化set
unordered_set<int> result; //存放最终结果
for (auto num : nums2) //c++11新标准 auto-自动推断类型,以及for的区间遍历
{
if (nums.find(num) != nums.end())
{//在nums 集合set中寻找是否有nums2的元素,找到返回该元素的迭代器,否则返回nums.end()
//找到说明num 两个数组中均有该元素
result.insert(num); //将其添加到结果集中
}
}
//因为返回类型是vector<int> , 所以需要返回一个已result初始化的匿名vector
return vector<int>(result.begin(), result.end());
}
};