leetcode 350. Intersection of Two Arrays II 解题小结

来源：互联网发布：站长seo查询编辑：程序博客网时间：2024/06/08 07:29

题目是这样子的:

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.

大意是，求出两个数组的交集，集合是无序的且不限重

这道题比较简单，不过关键是后期时间上的优化，一开始自己没往哈希方面想，直接来个暴力的两个for循环嵌套解法，复杂度O(n^2)可是击败了2%的人，无限汗颜中w(ﾟДﾟ)w，看了下dicuss发现有几个时间优化了挺多的解法。

第一种是先排序好两个数组，再利用迭代器进行遍历比较

class Solution {public:    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {      sort(nums1.begin(),nums1.end());      sort(nums2.begin(),nums2.end());      vector<int> a;      vector<int>::iterator s1 = nums1.begin(),s2= nums2.begin();      while(s1!=nums1.end()&&s2!=nums2.end())      {          while(s1!=nums1.end()&&s2!=nums2.end()&&*s1 == *s2)          {              a.push_back(*s1);              s1++;              s2++;          }          while(s2!=nums2.end()&&*s1>*s2)          {              s2++;          }          while(s1!=nums1.end()&&*s2>*s1)          {              s1++;          }      }    return a;    }};

第二种是利用了哈希:

a.没有用map，取两个数组的最大值作为长度开多一个数组进行哈希

class Solution {public:    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {      if(nums1.size()==0)      return nums1;      if(nums2.size()==0)      return nums2;      sort(nums1.begin(),nums1.end());      sort(nums2.begin(),nums2.end());      vector<int> b;      int max1 = max(nums1[nums1.size()-1],nums2[nums2.size()-1]);      vector<int> a(max1+1,0);      for(int i = 0; i<nums1.size();i++)      {          a[nums1[i]]++;      }      for(int i = 0 ;i<nums2.size();i++)      {        if(a[nums2[i]]!=0)        {            b.push_back(nums2[i]);            a[nums2[i]]--;        }      }      return b;    }};

b.利用了map进行哈希

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

c.与b大致相同，但是在if()中加入了预判断键值是否存在，减少了空间复杂度

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

0 0