力扣349-两个数组的交集-C++

一、运行结果

二、题目

给定两个数组 nums1 和 nums2 ，返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。

示例 1：

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

输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]
解释：[4,9] 也是可通过的

三、思路

题中说明不考虑结果的输出顺序，为方便进行操作，我们先将两个输入数组进行升序排序，然后开始对两个数组中的元素进行输入变量，遍历操作如下：

    ·若两个数组的当前元素相同，则查看结果数组ans，若结果数组为空或结果数组中的最后一个元素和两数组的当前元素不相等，则将该相等的元素插入到结果数组中，三个数组的索引下标均往后走一步；

    ·对比两个数组对应的当前元素，当期元素较小的数组索引往后走一步（因为数组已经排序，如果较小数组中有和较大元素相等的元素，则必在当前元素后面）。

四、代码

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ans;
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        int len1 = nums1.size();
        int len2 = nums2.size();
        // index1, index2作为遍历nums1 和 nums2的下标，index作为结果数组的下标
        int index1 = 0, index2 = 0, index = 0;  
        while(index1 < len1 && index2 < len2){
            if(nums1[index1] == nums2[index2]){
                if(ans.size()==0 || ans[index-1] != nums2[index2]){ //为了避免插入重复数据
                    ans.push_back(nums1[index1]);
                    index1++;
                    index2++;
                    index++;
                }else{
                    index1++;
                    index2++;
                }
            }
            else if(nums1[index1] < nums2[index2]){
                index1++;
            }
            else{
                index2++;
            }
        }
        return ans;
    }
};