leetcode349. 两个数组的交集

简单题，竟然想了20分钟= =

题目

给定两个数组 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

思路：

把两个vector分别放到set中（自动去重），然后遍历第一个set，如果在第二个set中可以找到该元素，就把它加入到交集数组中。

代码：

class Solution {
public:vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {// 思路：两个数组去重，然后遍历nums1，查询nums2中是否有对应的值set<int> st1, st2;int i = 0;for(i = 0; i < nums1.size(); i++){st1.insert(nums1[i]);}for(i = 0; i < nums2.size(); i++){st2.insert(nums2[i]);}vector<int> ans;for(set<int>::iterator it = st1.begin(); it!=st1.end(); it++){if(st2.find(*it) != st2.end()){ans.push_back(*it);}}return ans;}
};

运行结果：

知识点：

set的遍历，用迭代器it

在 set 中查找值，用 set.find(value)，它返回的值是迭代器，如果没有找到就是最后一个元素的后面即 st.end()

2023.8.19改进思路：

在遍历nums1的时候就放入哈希表中。然后再遍历nums2如果 hashTable[x] == 1就插入到vector中，再加加防止二次插入。

代码：

class Solution {
public:vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {int hashTable[1010] = {0};int i;for(i = 0; i < nums1.size(); i++){int x = nums1[i];if(hashTable[x] == 0){ // 防止二次++hashTable[x]++;}}vector<int> ans;for(i = 0; i < nums2.size(); i++){int x = nums2[i];if(hashTable[x] == 1){ // 防止往vector中插入两个相同的值ans.push_back(x);hashTable[x]++;}}return ans;}
};

运行结果：