两个数组的交集
力扣题目链接
我的解法:利用数组下标。
缺点:当取值范围很大时,浪费空间。
class Solution {
public:vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {int count1[1001]={0};int count2[1001]={0};for(int i=0;i<nums1.size();i++){count1[nums1[i]]++;}for(int i=0;i<nums2.size();i++){count2[nums2[i]]++;}int count =0;for(int i =0;i<1001;i++){if(count1[i]&&count2[i]){count++;}}vector<int> ret;ret.resize(count);int count3=0;for(int i =0;i<1001;i++){if(count1[i]&&count2[i]){ret[count3]= i;count3++;}}return ret;}
};
这道题目,主要要学会使用一种哈希数据结构:unordered_set,这个数据结构可以解决很多类似的问题。
注意题目特意说明:输出结果中的每个元素一定是唯一的,也就是说输出的结果的去重的, 同时可以不考虑输出结果的顺序
这道题用暴力的解法时间复杂度是O(n^2),那来看看使用哈希法进一步优化。
那么用数组来做哈希表也是不错的选择,例如242. 有效的字母异位词(opens new window)
但是要注意,使用数组来做哈希的题目,是因为题目都限制了数值的大小。
如果哈希值比较少、特别分散、跨度非常大,使用数组就造成空间的极大浪费。
此时就要使用另一种结构体了,set ,关于set,C++ 给提供了如下三种可用的数据结构:
- std::set
- std::multiset
- std::unordered_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> result_set; // 存放结果,之所以用set是为了给结果集去重unordered_set<int> nums_set(nums1.begin(), nums1.end());//将nums1的值插入哈希表for (int num : nums2) {// 发现nums2的元素 在nums_set里又出现过if (nums_set.find(num) != nums_set.end()) {result_set.insert(num);}}return vector<int>(result_set.begin(), result_set.end());}
};