C++Leetcode349:两个数组的交集

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

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

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

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

思路
1、暴力解法。两层遍历数组。count()函数很好用嘛~~时间复杂度为O(n^2)
2、哈希表unordered_set。遍历nums1将不重复的数存入一个无序哈希表s,然后再遍历nums2,将在数组和哈希表中,同时出现过一次的数存入另一个无序哈希表s1,最后将s1的值存入到目标数组res中。~这就是利用空间换时间呀,时间复杂度为O(n),n为nums1和nums2中较大的长度。
3、待优化。

实现方法
一、暴力解法

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        int s1=nums1.size(),s2=nums2.size();
        vector<int> result;
        for(int i=0;i<s1;i++){
            for(int j=0;j<s2;j++){
                if(nums1[i]==nums2[j] && count(result.begin(),result.end(),nums1[i])==0)
                    result.push_back(nums1[i]);
            }
        }
        return result;
    }
};

二、unordered_set

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> s,s1;
        vector<int> res;
        for(int k:nums1){
            if(s.count(k)==0)
                s.insert(k);  
        }
        for(int k:nums2){
            if(s.count(k)==1)
                s1.insert(k);
        }
        unordered_set<int>::iterator it=s1.begin();
        for(;it!=s1.end();++it){
            res.push_back(*it);
        }
        return res;
    }
};

版权声明:本文为weixin_43434305原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。