#350-两个数组的交集 II

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

示例 1:

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

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

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

分析:一开始的思路是,看看哪个vector的size大,然后逐一读取size小的元素,用find去检查,找的到就push_back进vector,然后erase,但是这个时间复杂度有点高。后来觉得,可以用元素的大小去限制一下
所以就先后排序,既然有了先后顺序,那就两个vector都一个一个读,相同就push_back和erase,哪一个元素大了,就让另外一个vector的指针往前走一步,直到任意一个vector走到了end()

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        sort(nums1.begin(),nums1.end());
        sort(nums2.begin(),nums2.end());
        
        vector<int>::iterator it1 = nums1.begin();
        vector<int>::iterator it2 = nums2.begin();
        
        vector<int> result;
        while(it1 != nums1.end() && it2 != nums2.end()){
            if(*it1 == *it2){
                result.push_back(*it1);
                it1 = nums1.erase(it1);
                it2 = nums2.erase(it2);
                
            }
            else if(*it1>*it2){
                it2++;
            }
            else if(*it1<*it2){
                it1++;
            }
            
        }
        return result;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值