两个数组的交集(C++实现)

附原题链接:https://leetcode-cn.com/problems/intersection-of-two-arrays

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

示例 1:

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

示例 2:

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

说明:

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

此题直接用暴力破解法,三个for()循环判断可解,但运行时间32ms,击败了个位数的用户。

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

后来看了大佬的做法,用哈希数据结构:unordered_set(set中每一个元素都是唯一的,没有重复。)运行时间8ms。

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> result_set; // 存放结果
        unordered_set<int> nums_set(nums1.begin(), nums1.end()); //用nums1初始化set
        for (int num : nums2) {
            if (nums_set.find(num) != nums_set.end()) { // 发现nums2的元素 在nums_set里又出现过
                result_set.insert(num);
            }
        }
        return vector<int>(result_set.begin(), result_set.end());
    }
};

 


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