LeeCode题目笔记

2019-11-14 两个数组的交集

2019-11-14  本文已影响0人  Antrn

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

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]

示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]

说明:

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

C++1

简单思路,
① 第一次遍历nums1找出拥有的不同的数据存放进hashhashset中。
② 第二次遍历nums2,如果某个元素在上一个hash表中存在,就将它放进另一个hashh中。
③ 最后遍历一次h,将里面的元素存进数组中,最后返回数组即可,这样保证数组中的数据是唯一的。

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> hashset;
        for(int i: nums1){
            if(hashset.count(i)==0){
                hashset.insert(i);
            }
        }
        unordered_set<int> h;
        vector<int> vect;
        
        for(int j:nums2){
            if(hashset.count(j)>0){
                h.insert(j);
            }
        }
        for ( auto it = h.begin(); it != h.end(); ++it )
            vect.push_back(*it);
        return vect;
    }
};
上一篇 下一篇

猜你喜欢

热点阅读