两个数据集的差集

【题目】

A、B两个整数集合，设计一个算法求他们的交集，尽可能的高效。

【来源】

腾讯2014校招软件开发笔试题目

【思路一】

对集合A和集合B进行排序（升序，用快排，平均复杂度O(N*logN)）。

设置两个指针p和q，同时指向集合A和集合B的第一个元素（已经升序排序）。用一个容器vector记录集合的交集。

如果两个元素大小相等的，说明是集合交集元素，加入交集vector中；

如果两个元素大小不相等的，p和q中较小值的指针，指向下一个元素。

依次操作，直到其中一个集合没有元素可比较为止。

排序：O(nlogn)

比较：O(n)

时间复杂度：O(n + nlogn + nlogn)

【代码一】

#include <iostream>
    #include <cstring>
    #include <vector>
    #include <algorithm>
    using namespace std;

    vector<int> IntersectionSet(vector<int> set1,vector<int> set2){
        int size1 = set1.size();
        int size2 = set2.size();
        // 排序
        sort(set1.begin(),set1.end());
        sort(set2.begin(),set2.end());
        int i = 0,j = 0;
        vector<int> result;
        // 比较
        while(i < size1 && j < size2){
            // 相同的加入差集中
            if(set1[i] == set2[j]){
                result.push_back(set1[i]);
                ++i;
                ++j;
            }//if
            // 值小的跳过
            else if(set1[i] < set2[j]){
                ++i;
            }//else
            else{
                ++j;
            }//else
        }//while
        return result;
    }

    int main(){
        vector<int> set1 = {1,22,6,9,11,5};
        vector<int> set2 = {2,10,11,4,22,3,15};
        vector<int> result = IntersectionSet(set1,set2);
        for(int i = 0;i < result.size();++i){
            cout<<result[i]<<" ";
        }//for
        cout<<endl;
        return 0;
    }