309、两个数组的交集

题目描述：

给定两个数组 nums1 和 nums2 ，返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
示例 1:

输入：nums1 = [1,2,2,1], nums2 = [2,2]
输出：[2]

示例 2:

输入：nums1 = [1,2,2,1], nums2 = [2,2]
输出：[2]

提示：

• 1 <= nums1.length, nums2.length <= 1000
• 0 <= nums1[i], nums2[i] <= 1000

解题思路

一开始我的思路是，分别开两个容量为1000的数组a1和a2用来统计两个数组的值出现的次数，然后遍历一遍统计好的数组，如果a1和a2数组在同一个下标下都不为0，则表示nums1和nums2都出现了值为这个数组下标的元素，就把它放进vector容器中。

但这种方法的弊端是，我们不知道所给的数组nums1和nums2的大小，如果统一都用大小为1000的数组统计就很容易造成内存浪费，这里可以优化成unordered_set存放。

优化前代码

class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
	int len1=nums1.size(),len2=nums2.size();
	int a1[1001]={0},a2[1001]={0};
	for(int i=0;i<len1;i++){
		a1[nums1[i]]++;
	}
	for(int i=0;i<len2;i++){
		a2[nums2[i]]++;
	}

	vector<int>v;
	for(int i=0;i<=1000;i++){
		if(a1[i]&&a2[i])
			v.push_back(i);
	}
	return v;
}
};

优化后代码

class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
	unordered_set<int> result_set;//存放结果
	unordered_set<int> set(nums1.begin(),nums1.end());//将nums1数组转换为无序容器
	for(int num:nums2){
		if(set.find(num)!=set.end()){
			result_set.insert(num);
		}
	}
	return vector<int>(result_set.begin(),result_set.end());
	}
};