LintCode-[容易] 547. 两数组的交

2017-08-23 12:21:34来源:CSDN作者:Cai_Guanglei人点击

分享

描述:

返回两个数组的交

注意事项:

Each element in the result must be unique.
The result can be in any order.
(在输出里的每个元素必须唯一,输出里元素的顺序不作要求)

样例:

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

思路:

先用sort函数将两个向量按从小到大排序,再删除重复的数字。然后从头同时遍历两个向量,如果在同一下标处两个数相等则加到结果向量中,两向量下标同时加一,如果a > b, 则b所在的向量下标加一。

C++实现:

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>result;        int i1 = 0;        int i2 = 0;        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 size1 = nums1.size();        int size2 = nums2.size();        while (i1 < size1 && i2 < size2) {            if (nums1[i1] == nums2[i2]) {                result.push_back(nums1[i1]);                i1++;                i2++;            }            else if (nums1[i1] > nums2[i2]) {                i2++;            }            else {                i1++;            }        }        return result;    }};

最新文章

123

最新摄影

微信扫一扫

第七城市微信公众平台