蓝桥杯 算法训练 集合运算

蓝桥杯 集合运算

问题描述

　　给出两个整数集合A、B，求出他们的交集、并集以及B在A中的余集。

输入格式

　　第一行为一个整数n，表示集合A中的元素个数。
　　第二行有n个互不相同的用空格隔开的整数，表示集合A中的元素。
　　第三行为一个整数m，表示集合B中的元素个数。
　　第四行有m个互不相同的用空格隔开的整数，表示集合B中的元素。
　　集合中的所有元素均为int范围内的整数，n、m<=1000。

输出格式

　　第一行按从小到大的顺序输出A、B交集中的所有元素。
　　第二行按从小到大的顺序输出A、B并集中的所有元素。
　　第三行按从小到大的顺序输出B在A中的余集中的所有元素。

样例输入

5
1 2 3 4 5
5
2 4 6 8 10

样例输出

2 4
1 2 3 4 5 6 8 10
1 3 5

样例输入

4
1 2 3 4
3
5 6 7

样例输出

1 2 3 4 5 6 7
1 2 3 4

 

这道题可以直接使用C++ stl里面自带的函数。

set_intersection求交集，set_union求并集，set_difference求差集。注意，在这里要求vector集合是有序的，因此需要先对v1和v2进行排序。

#include <bits/stdc++.h>

using namespace std;

int main()
{
    int n,m,i;
    int val;
    vector<int> v1;
    vector<int> v2;
    vector<int> v_intersection;
    vector<int> v_union;
    vector<int> v_difference;
    cin>>n;
    for(i=0;i<n;i++){
        scanf("%d",&val);
        v1.push_back(val);
    }
    cin>>m;
    for(i=0;i<m;i++){
        scanf("%d",&val);
        v2.push_back(val);
    }
    sort(v1.begin(),v1.end());
    sort(v2.begin(),v2.end());
    //求交集
    set_intersection(v1.begin(),v1.end(),v2.begin(),v2.end(),back_inserter(v_intersection));
    if(v_intersection.size()!=0){
        for(i=0;i<v_intersection.size();i++){
            cout<<v_intersection[i]<<" ";
        }
        cout<<endl;
    }
    //求并集
    set_union(v1.begin(),v1.end(),v2.begin(),v2.end(),inserter(v_union,v_union.begin()));
    for(i=0;i<v_union.size();i++){
        cout<<v_union[i]<<" ";
    }
    cout<<endl;
    //求差集
    set_difference(v1.begin(),v1.end(),v2.begin(),v2.end(),inserter(v_difference,v_difference.begin()));
    for(i=0;i<v_difference.size();i++){
        cout<<v_difference[i]<<" ";
    }
    return 0;
}