代码随想录13

349. 两个数组的交集

已解答

简单

相关标签

相关企业

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

示例 1：

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

示例 2：

输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]
解释：[4,9] 也是可通过的

提示：

• 1 <= nums1.length, nums2.length <= 1000

• 0 <= nums1[i], nums2[i] <= 1000

用Set写吧 我有点唐了

#pragma region 两个数组的交集
class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
		vector<int> result;
		map<int, int> countMap;
		map<int, int> resultMap;
		for (size_t i = 0; i < nums1.size(); i++)
		{
			if (countMap.find(nums1[i]) == countMap.end()) {
				countMap[nums1[i]] = 1;
			}
			else {
				countMap[nums1[i]]++;
			}
		}
		//countMap此时存储的是nums1的数字和数量 且没有重复数字
		//此时对第二个数组进行遍历
		for (size_t i = 0; i < nums2.size(); i++)
		{
			if (countMap.find(nums2[i]) != countMap.end()) {
				if (resultMap.find(nums2[i]) == resultMap.end()) {
					//如果resultMap中没有这个数字 则将其加入
					resultMap[nums2[i]] = 1;
					result.push_back(nums2[i]);
				}
			}
		}

		//此时result存储的是两个数组的交集
		return result;
    }
};
#pragma endregion