代码题(21)— 两数组交集、两数组交集II

1、349. 两个数组的交集

给定两个数组,写一个函数来计算它们的交集。

例子:

 给定 num1[1, 2, 2, 1]nums2 = [2, 2], 返回 [2].

提示:

  • 每个在结果中的元素必定是唯一的。
  • 我们可以不考虑输出结果的顺序。

  这道题让我们找两个数组交集的部分(不包含重复数字),难度不算大,我们可以用个set把nums1都放进去,然后遍历nums2的元素,如果在set中存在,说明是交集的部分,加入结果的set中,最后再把结果转为vector的形式即可:

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        if(nums1.empty() || nums2.empty())
            return res;
        unordered_set<int> s(nums1.begin(), nums1.end());
        unordered_set<int> temp;
        for(int val : nums2)
        {
            if(s.count(val))
                temp.insert(val);
        }
        return vector<int>(temp.begin(), temp.end());
    }
};

2、350. 两个数组的交集 II

给定两个数组,写一个方法来计算它们的交集。

例如:
给定 nums1 = [1, 2, 2, 1]nums2 = [2, 2], 返回 [2, 2].

注意:

    •    输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
    •    我们可以不考虑输出结果的顺序。

这道题是之前那道Intersection of Two Arrays的拓展,不同之处在于这道题允许我们返回重复的数字,而且是尽可能多的返回,之前那道题是说有重复的数字只返回一个就行。那么这道题我们用哈希表来建立nums1中字符和其出现个数之间的映射, 然后遍历nums2数组,如果当前字符在哈希表中的个数大于0,则将此字符加入结果res中,然后哈希表的对应值自减1,

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        if(nums1.empty() || nums2.empty())
            return res;
        unordered_map<int, int> m;
        for(int val : nums1)
            m[val]++;
        for(int val : nums2)
        {
            if(m[val] > 0)
            {
                m[val]--;
                res.push_back(val);
            }
                
        }
        return res;
    }
};

 

posted @ 2018-07-22 15:18  深度机器学习  阅读(314)  评论(0编辑  收藏  举报