350. Intersection of Two Arrays II & Add to List 349. Intersection of Two Arrays 笔记

  1. 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。
代码如下:

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int, int> m;
        vector<int> res;
        for(int i = 0; i < nums1.size(); i++)
        {
            m[nums1[i]]++;
        }
        for(int i = 0; i < nums2.size(); i++)
        {
            m[nums2[i]]--;
            if(m[nums2[i]] >= 0)
                res.push_back(nums2[i]);
        }
        return res;
    }
};
  1. Intersection of Two Arrays
    和上边的区别是重复的元素只输出一次。
    就在遍历第二个数组的时候,输出这个之后把哈希表里这个
    数的值置为0就行了。
    代码如下:
class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int, int> m;
        vector<int> res;
        for(int i = 0; i < nums1.size(); i++)
        {
            m[nums1[i]]++;
        }
        for(int i = 0; i < nums2.size(); i++)
        {
            m[nums2[i]]--;
            if(m[nums2[i]] >= 0)
            {
                res.push_back(nums2[i]);
                m[nums2[i]] = 0;
            }
        }
        return res;
        }
};
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容