LeetCode解题报告 349. Intersection of Two Arrays [easy]

题目描述

Given two arrays, write a function to compute their intersection.

Example:
Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2].

Note:

• Each element in the result must be unique.
• The result can be in any order.

解题思路

题目要求求两个数组的交集，并且交集中的元素不重复。

这道题是在sort分类中选择的，因此考虑用排序的方法来实现。

将两个数组都先排成有序数列，从小到大，用两个指针来表示当前所指到元素在数列中的位置。如果指的两个数大小不相同，则根据大小向后移动其中的一个指针，移动数字较小的那一个；如果两个数字相等，并且这个数字与结果数组中的最后一个数不想等时，则说明没有重复，则将这个数字加入到结果数组中。需要注意的是，结果数组刚开始为空，因此第一次的判断应该是当数组为空时，就将相等的这个数加入，不需要满足“这个数字与结果数组中的最后一个数不想等”这个条件。

复杂度分析

排序所需要的复杂度为O(N*logN)，移动指针比较需要时间复杂度为O(N)，总复杂度也为O(N*logN)

代码如下：

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        vector<int>res;
        int i=0,j=0;
        while (i<nums1.size()&&j<nums2.size()) {
            if(nums1[i]>nums2[j]){
                j++;
            }
            else if (nums1[i]<nums2[j]){
                i++;
            }
            else{
                if (res.size()==0||nums1[i]!=res.back()){
                    res.push_back(nums1[i]);
                }
                i++;
                j++;
            }
        }
        return res;
    }
};