求兩個整數集合的交集-2014TX筆試題

假設整數爲INT型，且非負數，代碼以下：

注意點：GetInersection返回值應該爲vector<int>而不該該是vector<int>& 由於vector<int>向量對象在函數運行完後系統會

自動的調用析構函數，若是返回&的話，將獲得一個NULL值

#include <iostream>
#include <bitset>
#include <set>
#include<vector>
using namespace std;

//求集合a和集合b的交集
vector<int> GetIntersection(set<int>a,set<int>b)
{
    bitset<2^32> bs(0);//建立一個位數組來存儲a中存在的數
    vector<int >c;//建立向量來存儲交集中的元素

    //把a中存在的數值做爲c的下標，標記爲存在
    for(set<int>::iterator iter=a.begin ();iter!=a.end();iter++)
    {
        bs[*iter]=1;
    }

    //判斷b中的元素是否a中存在，存在的話加入向量c中
    for(set<int>::iterator iter=b.begin ();iter!=b.end();iter++)
    {
        if(bs[*iter]==1)c.push_back(*iter);
    }
    return c;
}

void main(){
    //計算兩個集合的交集
    set<int> a;
    set<int> b;
    for(int i=0;i<10;i++){a.insert(i);b.insert(i*2);}

    vector<int >c=GetIntersection(a,b);
    for(vector<int>::iterator iter=c.begin();iter!=c.end();iter++){
        cout<<*iter;
    }

    system("pause");
}