两数组的交

两数组的交

问题描述：

返回两个数组的交。

样例

nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2].

解题思路：

先对给定的两个数组排序，在利用unique（）函数去重，最后遍历两数组比较值。

代码实现：

class Solution {
public:
    /**
     * @param nums1 an integer array
     * @param nums2 an integer array
     * @return an integer array
     */
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        // Write your code here
        vector<int>::iterator it1;
        vector<int>::iterator it2;
        vector<int>r;
        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());//去掉相邻重复元素
        int i=0;
        int j=0;
        while(i<nums1.size()&&j<nums2.size()){
            if(nums1[i]==nums2[j]){
                r.push_back(nums1[i]);
                i++;
                j++;
            }
            if(nums1[i]<nums2[j])  i++;
            if(nums1[i]>nums2[j])  j++;
        }
        return r;
    }
};

A题感悟：

此题的思路很容易想到，主要是如何去重和怎样解决时间复杂度的问题。一开始我是直接遍历两个数组，边找两数组的交边去重，结果只能通过84%的数据，超过了要求的时间复杂度的限制，因此只能另辟蹊径。在ACM程序设计的书中有一个去重函数unique（），但是只能去掉相邻的相同元素，所以要先进行排序。排序不仅对去重函数起了作用，更对后面的两数组比较值起了大作用，使得不需要使用双重循环就可以比较值。