题目
- 两个数组的交集
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
说明:
输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。
哈希法
不能再使用数组作为哈希表,因为数组作哈希表不能太大(如作为字母表),此题没有限定数值范围,不用数组。
当我们要使用集合解决哈希问题时,优先使用unordered_set.因为它的查询和增删效率最高。unordered_set是无序、无重复数据的。
如果集合有序,就用set。
如果集合有序且有重复数据,用multiset。
题目中要求的集合是无序且无重复数据的,用unordered_set最好。
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int> num(nums1.begin(), nums1.end());
unordered_set<int> result;
for (int i = 0; i < nums2.size(); i++) {
if (num.find(nums2[i]) != num.end()) {
result.insert(nums2[i]);
}
}
return vector<int> (result.begin(), result.end());
}
};