Intersection of Two ArraysI/II题解

Intersection of Two ArraysI

https://leetcode.com/problems/intersection-of-two-arrays/

这道题不是很难，我用unordered_set，存储nums1中的数据，和nums1和nums2中相同的数据

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        if(nums1.size()==0||nums2.size()==0) return res;
        unordered_set<int> num1;
        unordered_set<int> res1;
        for(int i=0;i<nums1.size();i++){
            if(num1.find(nums1[i])!=num1.end()) continue;
            else num1.insert(nums1[i]);
        }
        for(int i=0;i<nums2.size();i++){
            if(num1.find(nums2[i])!=num1.end()){
                if(res1.find(nums2[i])==res1.end()){
                    res1.insert(nums2[i]);
                    res.push_back(nums2[i]);
                } 
                else continue;
            }
            else continue;
        }
        return res;
    }
};

Intersection of Two ArraysII

https://leetcode.com/problems/intersection-of-two-arrays-ii/

这道题和前一道题不同的地方就是要保证每一个重复的数字都要记录，set这个数据结构中存储的每一个元素都是不同的，所以这道题不能用set来存储数据。这个时候我们可以考虑，既然要存储每一个出现过的数，可以对两个数组进行排序，排序之后在进行遍历，当两个数字相等就存入数组。

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        if(nums1.size()==0||nums2.size()==0) return res;
        int i=0,j=0;
        sort(nums1.begin(),nums1.end());
        sort(nums2.begin(),nums2.end());
        while(i<nums1.size()&&j<nums2.size()){
            if(nums1[i]<nums2[j]) i++;
            else if(nums1[i]>nums2[j]) j++;
            else{
                res.push_back(nums1[i]);
                i++;j++;
            }
        }
        return res;
    }
};

在leetcode的discuss上还看了一个用map的：

class Solution {
public:
  vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
    vector<int> ret;
    unordered_map<int, int> umap;
    
    for (auto i : nums1)
    {
      umap[i] ++;
    }

    for (auto j : nums2)
    {
      if (umap.find(j) != umap.end() && umap[j] > 0)
      {
	ret.push_back(j);
	umap[j] --;
      }
    }
    return ret;
  }
};