【蓝桥杯】ALGO-10 集合运算

正文索引 [隐藏]

题目描述:

给出两个整数集合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

解题思路:

这题set就完事了,主要考察了set的三个函数:①set_intersection求集合的交集,②set_union求集合的并集,③set_difference求集合的差集。这三个函数都有5个参数:集合A的begin、end,集合B的begin、end,还有一个back_inserter()函数,这是iterator适配器,它能使元素被插入到作为实参的某种容器的尾部。提交直接AC击掌nice!

AC代码:

#include <bits/stdc++.h>
using namespace std;
#define Up(i,a,b) for(int i = a; i <= b; i++)

void print(vector<int> &v)  //打印vector
{
    int n = v.size()-1;
    Up(i,0,n)
    {
        printf("%d%s",v[i],i==n?"\n":" ");
    }
}

int main()
{
    ios::sync_with_stdio(false);
    int n;
    cin >> n;
    set<int> A,B; 
    Up(i,1,n)
    {
        int _;
        cin >> _;
        A.insert(_); 
    }
    cin >> n;
    Up(i,1,n)
    {
        int _;
        cin >> _;
        B.insert(_);
    }
    vector<int> v1,v2,v3;  //交集v1,并集v2,差集v3
    set_intersection(A.begin(),A.end(),B.begin(),B.end(),back_inserter(v1));  //求A、B的交集
    print(v1);
    set_union(A.begin(),A.end(),B.begin(),B.end(),back_inserter(v2));   //求A、B的并集
    print(v2);
    set_difference(A.begin(),A.end(),B.begin(),B.end(),back_inserter(v3));    //求A、B的差集
    print(v3);
    return 0;
}