数组交集

[349. 两个数组的交集](#349. 两个数组的交集)
[350. 两个数组的交集Ⅱ](#350. 两个数组的交集Ⅱ)
- [排序 + 双指针](#排序 + 双指针)
- 数组实现哈希表
- unordered_map

349. 两个数组的交集

题目内容如下，理解题意：①交集中每个元素是唯一的，只能出现一次，所以本题要找的是同时出现在数组nums1和nums2中的元素 ，但是并不关心他们出现的次数 ；②输出结果的顺序不用考虑，也就是只要找到了同时出现在nums1和nums2中的元素就行，可以给数组排序（打乱了原本的顺序）再去查找、可以用map、set（其中key是无序的）......

这个题目暴力求解是可以的，暴力求解两层循环，将nums1和nums2的元素逐个对比，时间复杂度是O(m*n)，因为nums1和nums2的长度都<=1000，所以最终也是能够运行的。

考虑更优的解法，直接一遍遍历nums1和nums2就好了。以下详述各解法：

排序＋双指针

解法Ⅰ，对nums1和nums2排序 后，从头开始遍历两个数组（下标用index1和index2），并将nums1[index1] = nums2[index2]的元素加入结果数组中。

存在的问题是：因为nums1和nums2中存在重复元素，如果找到了nums1[index1] = nums2[index2]，且在nums1中，nums1[index1] 有重复，即nums1[index1+1] = nums1[index1] ，且在nums2中，nums2[index2]有重复，即nums2[index2+1] = nums2[index2]。那么直接对index++和index2++，会向结果数组中添加重复的元素。

解决：找到nums1[index1] = nums2[index2]后，index1++直到找到与之不同的下一个元素（就是跳过中间的相同的元素）；index2++同样。

代码实现（C++）：

cpp 复制代码

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ans;
        sort(nums1.begin(),nums1.end());//先给nums1和nums2都排序
        sort(nums2.begin(),nums2.end());
        //排序后双指针（index1和index2遍历nums1和nums2
        for(int index1 = 0, index2 = 0; index1 < nums1.size() && index2 < nums2.size();){
            if(nums1[index1] == nums2[index2]){
                ans.emplace_back(nums1[index1]);    //如果找到在两个数组中出现的元素，加入到结果数组中
                //之后直接跳过和当前元素相同的一截，避免有可能向ans中重复添加该元素的可能
                do{
                    index1++;
                }while(index1 < nums1.size() && nums1[index1] == nums1[index1 - 1]);
                do{
                    index2++;
                }while(index2 < nums2.size() && nums2[index2] == nums2[index2 - 1]);
            }
            //如果不相等，更小的那个数向后移，同样是跳过和当前元素相同的一截，避免重复的比较
            else  if( nums1[index1] < nums2[index2]){
                do{
                    index1++;
                }while(index1 < nums1.size() && nums1[index1] == nums1[index1 - 1]);                
            }
            else{
                do{
                    index2++;
                }while(index2 < nums2.size() && nums2[index2] == nums2[index2 -1]);                
            }
        }
        return ans;
    }
};

数组实现哈希表

哈希表的好处体现在，它能够快速查找一个元素是否存在，时间复杂度是O(1)。我们要找nums1和nums2中同时存在的元素，换句话------查找nums1中某个元素是否出现在了nums2中。那么就可以用哈希表。因为题目中，nums1和num2的长度以及其中的int元素的大小都给了限制（<=1000），那么可以用数组来实现哈希表。

直接定义长度为1001的int数组count_1和count_2 ，统计nums1中元素的次数和nums2中元素出现的次数，最后对比count_1和count_2的每位元素，如果同时不为0 的话，将对应元素加入到ans中。

代码如下（C++）：

cpp 复制代码

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        int count_1[1001] = {0}, count_2[1001] = {0};
        vector<int> ans;
        //分别统计nums1和nums2中元素出现情况
        for ( auto& num : nums1){
            count_1[num]=1;
        }
        for ( auto& num : nums2){            
            count_2[num]++;
        }
        for ( int i = 0; i <= 1000; i++){
        	//在两个数组中出现次数均>=1时，加入ans中
            if( count_1[i] && count_2[i])
                ans.emplace_back(i);
        }
        return ans;
    }
};

优化：上面需要用到两个数组count_1和count_2来分别统计nums1、nums2中元素出现的情况，之后还要遍历这俩数组。是否有可能只使用一个count数组，用两次 ？------遍历nums1的时候，出现的元素不统计次数，而是count[nums[i]]=1，标记该元素出现过；遍历nums2的时候，如果count[nums2[j]] !=0就表示在两个数组中同时存在；防止重复添加，再将count[nums2[j]]=0;

代码实现（C++）：

cpp 复制代码

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        int count[1001] = {0};
        vector<int> ans;
        //统计nums1中元素的出现情况
        for ( auto& num : nums1){
            count[num]=1;
        }
        //遍历nums2
        for ( auto& num : nums2){
            if(count[num]){ //同时判断nums2中的元素在nums1中是否出现
                count[num]=0;
                ans.emplace_back(num);
            }
        }      
        return ans;
    }
};

unordered_set

题意是要找交集，那么直接把数组变成集合，然后求两个集合交集就好。实现上，将vector变成unordered_set，然后对比两个unordered_set的key，找到重叠部分。

代码实现（C++）：

cpp 复制代码

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
    	//把vector转化成unordered_set
        unordered_set<int> num_set1(nums1.begin(),nums1.end());
        unordered_set<int> num_set2(nums2.begin(),nums2.end());
        vector<int> ans;
        //遍历两个set，找交集；遍历size小的
        if( num_set1.size() < num_set2.size()){
            for( auto& key : num_set1)
                if(num_set2.count(key))
                    ans.emplace_back(key);
        }
        else{
            for( auto& key :num_set2)
                if(num_set1.count(key))
                    ans.emplace_back(key);
        }
        return ans;
    }
};

unordered_map

最后也能用map来实现，遍历nums1和nums2的同时，统计其中元素出现的次数，用unordered_map来存，key是数组中出现的元素，value是元素出现的次数；之后找到两个map中重合的key。

代码（C++）：

cpp 复制代码

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int,int> count_1, count_2; 
        vector<int> ans;
        //用map统计数组中出现的元素及其次数
        for ( auto& num : nums1){
            count_1[num]++;
        }
        for ( auto& num : nums2){
            count_2[num]++;
        }
        if(count_1.size() < count_2.size()){
            for ( auto key_value : count_1)
                if(count_2.count(key_value.first))
                    ans.emplace_back(key_value.first);
        }
        else{
            for ( auto key_value : count_2)
                if(count_1.count(key_value.first))
                    ans.emplace_back(key_value.first);
        }
        return ans;
    }
};

350. 两个数组的交集Ⅱ

题目链接：350. 两个数组的交集Ⅱ

题目内容：

这道题和上一题唯一不同的点在于：在nums1和nums2中同时出现的元素，如果出现次数不止一次，都需要加入到结果中。即不仅要统计同时出现在nums1和nums2中的元素，还要统计他们各自出现的次数（C1和C2），并在结果数组ans中重复min (C1, C2) 次。以下代码均在上一题基础上做一点改动即可。

排序 + 双指针

排序后数组元素逐个对比就好：双指针index1和index2，每次对比nums1[index1]和nums2[index2]的关系后，直接index1++，index2++：

cpp 复制代码

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ans;
        sort(nums1.begin(),nums1.end());
        sort(nums2.begin(),nums2.end());
        for(int index1 = 0, index2 = 0; index1 < nums1.size() && index2 < nums2.size();){
            if(nums1[index1] == nums2[index2]){
                ans.emplace_back(nums1[index1]);
                //下标直接向后移动，这样同时重复出现在两个数组中的元素能够重复添加
                index1++;
                index2++;               
            }
            else  if( nums1[index1] < nums2[index2]){
                index1++;              
            }
            else{
                index2++;               
            }
        }
        return ans;
    }
};

数组实现哈希表

先用数组count统计nums1中出现的元素，及其次数；再遍历nums2，同时出现在nums1中的元素，count[nums2[i]]- -，向结果数组ans中添加一次该元素。

cpp 复制代码

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        int count[1001] = {0};
        vector<int> ans;
        //统计出现的元素，以及次数
        for ( auto& num : nums1){
            count[num]++;
        }
        for ( auto& num : nums2){
            if(count[num]){
            	//对于同时出现的元素，对其次数--，保证后续再出现该元素时，还能重复添加
                count[num]--;
                ans.emplace_back(num);
            }          
        }      
        return ans;
    }
};

unordered_map

只是将上面的数组换成了unordered_map。用数组实现哈希表适用于nums1和nums2都不大的，并且其中元素也不大的情况，当数组很大并且数组元素为int，大小没有限制的时候，用map更合适（或者set）。

代码如下：

cpp 复制代码

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int,int> count;
        vector<int> ans;
        for(auto& num : nums1){
            count[num]++;
        }
        for(auto& num : nums2){
            if(count.count(num)){
                ans.emplace_back(num);
                count[num]--;
                //如果已经重复添加了min(c1,c2)次了，即便后续再出现也不能再添加了
                if(count[num]==0)
                    count.erase(num);
            }
        }
        return ans;
    }
};

【leetcode 力扣刷题】数组交集（数组、set、map都可实现哈希表）

数组交集

349. 两个数组的交集

排序＋双指针

数组实现哈希表

unordered_set

unordered_map

350. 两个数组的交集Ⅱ

排序 + 双指针

数组实现哈希表

unordered_map