LeetCode 349. Intersection of Two Arrays && 350. Intersection of Two Arrays II

最新推荐文章于 2024-11-01 15:19:21 发布

yurio7r

最新推荐文章于 2024-11-01 15:19:21 发布

阅读量279

点赞数

文章标签： leetcode

本文链接：https://blog.csdn.net/MrSiz/article/details/51820724

版权

LeetCode 专栏收录该内容

178 篇文章 0 订阅

订阅专栏

描述

两道题目都是求交集，前者要求最后的结果集中只存在一个相同的值，后者要求尽结果集中应该有尽可能多的相同值（以共同值较少的数量为准）。

解决

349直接暴力遍历的，复杂度为O(n^2)。

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ans;
        map<int, int> m;
        auto nums1_end = nums1.end();
        auto nums2_end = nums2.end();
        for (auto it = nums1.begin(); it != nums1_end; ++it){
            if (m[*it] == 0){
                for (auto zz = nums2.begin(); zz != nums2_end; ++zz){
                    if (*zz == *it && m[*it] == 0){
                        ans.push_back(*it);
                        m[*it] = 1;
                    }
                }
            }
        }
        return ans;
    }
};

350就是两个集合同时向前遍历，复杂度为O(n)。

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        auto it1 = nums1.begin();
        auto it2 = nums2.begin();
        while (it1 != nums1.end() && it2 != nums2.end()){
            while (it1 != nums1.end() && it2 != nums2.end() && *it1 == *it2){
                res.push_back(*it1);
                it1++, it2++;
            }
            while (it1 != nums1.end() && it2 != nums2.end() && *it1 < *it2){
                it1++;
            }
            while (it1 != nums1.end() && it2 != nums2.end() && *it1 > *it2){
                it2++;
            }
        }
        return res;
    }
};