力扣 两个数组的交集

力扣 两个数组的交集

题目描述

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

示例 1：

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

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

说明：

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

要点分析

将数组排序后去重,然后遍历数组找出共同元素即可

难点在于unique函数, sort函数以及erase函数的应用

unique函数: 将相邻的元素中重复的部分移动到vector的尾部

erase函数: 去除元素

sort函数: 进行排序

AC代码:

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        //传入的是首地址和尾地址
        sort(nums1.begin(),nums1.end());
        sort(nums2.begin(),nums2.end());
        nums1.erase(unique(nums1.begin(), nums1.end()), nums1.end());
        nums2.erase(unique(nums2.begin(),nums2.end()),nums2.end());
        vector<int>ans;
        for(int i=0,j=0;i<nums1.size()&&j<nums2.size();)
        {
            if(nums1[i]==nums2[j])
            {
                ans.push_back(nums1[i]);
                i++;
                j++;
            }
            else if (nums1[i]<nums2[j])
            {
                i++;
            }
            else if (nums1[i]>nums2[j])
            {
                j++;
            }
        }
        return ans;
    }
};