难度简单311
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2] 输出:[2]示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出:[9,4]
说明:
- 输出结果中的每个元素一定是唯一的。
- 我们可以不考虑输出结果的顺序。
用哈希表存储数组,然后去求交集。
需要掌握的是,这题为什么用哈希表去存储数组,这样做的理由是什么?是这样的:
哈希表存储的元素查找的时间复杂度在O(1)
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int> set1, set2;
for(auto& num: nums1){
set1.insert(num);
}
for(auto& num: nums2){
set2.insert(num);
}
return getIntersection(set1, set2);
}
vector<int> getIntersection(unordered_set<int>& set1, unordered_set<int>& set2){
vector<int> res;
for(auto& num: set1){
if(set2.count(num)){
res.push_back(num);
}
}
return res;
}
};
然后还有一些其他的需要总结,
这里用set来存储数组,即使数组里包含重复元素,但在本题仍然可以用set存储是因为本题只关注两个数组的交集,所以数组中重复元素不无影响。同时set存储降低查找的空间和时间复杂度。
还有一个是,参数的类型需要保持一致,这个与写算法时的专注度有关。自己写着写着走神了,回头接着时候没有认真思考好就容易写错,这样的错误还不好找。所以,学习时还是要调整好状态,专注学习!
加油!