350. Intersection of Two Arrays II

Given two arrays, write a function to compute their intersection.

Example:
Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2, 2].

Note:
Each element in the result should appear as many times as it shows in both arrays.
The result can be in any order.

题解

找两个数组的交集，要求可以重复，无需顺序。

1. Hash表
   时间复杂度O(m+n), 空间复杂度O(m), 其中m为nums1的大小，n为nums2的大小
   如果明确知道m和n数量级相差较大，如m很大，就通过nums2建Hash表，减小空间复杂度，时间复杂度仍为O(m+n)。

   class Solution {
   public:
       vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
           unordered_map<int, int> haha;
           vector<int> ret;
           for(int i = 0; i < nums1.size(); i++){
               haha[nums1[i]]++;
           }
           for(int i = 0; i < nums2.size(); i++){
               if(haha.count(nums2[i]) && haha[nums2[i]]-- > 0){
                   ret.push_back(nums2[i]);
               }
           }
           return ret;
       }
   };

   你当然也可以这样写,但是空间复杂度就是O(m+n)。原因是使用下标访问时，如果key不存在，会自动为其赋一个默认值。 要避免只需加一个find或count的判断，推荐使用count。

   class Solution {
   public:
       vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
           unordered_map<int, int> haha;
           vector<int> res;
           for(int i = 0; i < nums1.size(); i++) haha[nums1[i]]++;
           for(int i = 0; i < nums2.size(); i++)
               if(haha[nums2[i]]-- > 0) res.push_back(nums2[i]);
           return res;
       }
   };

2. sort+双指针

   时间复杂度O(max(m, n) log(max(m, n)))，空间复杂度为O(m+n) （不要忽略sort）

   当数组已有序时时间复杂度为O(min(m,n))

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