OJ 1349 Problem B STL——集合运算

Description

集合的运算就是用给定的集合去指定新的集合。设A和B是集合，则它们的并差交补集分别定义如下：
A∪B={x|x∈A∨x∈B}
A∩B={x|x∈A∧x∈B}
A-B={x|x∈A∧x不属于 B}
SA ={x|x∈(A∪B)∧x 不属于A}
SB ={x|x∈(A∪B)∧x 不属于B}

Input

第一行输入一个正整数T，表示总共有T组测试数据。（T<=200）
然后下面有2T行，每一行都有n+1个数字，其中第一个数字是n(0<=n<=100)，表示该行后面还有n个数字输入。

Output

对于每组测试数据，首先输出测试数据序号，”Case #.NO”，
接下来输出共7行，每行都是一个集合，
前2行分别输出集合A、B，接下5行来分别输出集合A、B的并(A u B)、交(A n B)、差(A – B)、补。
集合中的元素用“{}”扩起来，且元素之间用“， ”隔开。

Sample Input

1
4 1 2 3 1
0

Sample Output

Case# 1:
A = {1, 2, 3}
B = {}
A u B = {1, 2, 3}
A n B = {}
A - B = {1, 2, 3}
SA = {}
SB = {1, 2, 3}

HINT

如果你会用百度搜一下关键字“stl set”，这个题目我相信你会很快很轻松的做出来。加油哦！

Accepted Code

#include <iostream>
#include <vector>
#include <map>
#include <stack>
#include <queue>
#include <list>
#include <algorithm>
#include <string>
#include <set>
#include <cstdio>
using namespace std;
void print(set<int> &A) {
    set<int>::iterator iter;
    cout << "{";
    for(iter = A.begin(); iter != A.end(); ++iter)
        if (iter == A.begin()) cout << *iter;
        else cout << ", " << *iter;
    cout << "}" << endl;
}
int main() {
    int T;
    cin >> T;
    for(int flag = 1; flag <= T; flag++) {
        int t;
        cin >> t;
        set<int> A, B;
        set<int> s1, s2, s3, s4, s5;
        for(int i = 0; i < t; i++) {
            int f; cin >> f;
            A.insert(f);
        }
        cin >> t;
        for(int i = 0; i < t; i++) {
            int f; cin >> f;
            B.insert(f);
        }
        printf("Case# %d:\n", flag);
        cout << "A = "; print(A);
        cout << "B = "; print(B);
        set_union(A.begin(), A.end(), B.begin(), B.end(), inserter(s1, s1.begin()));
        cout << "A u B = "; print(s1);
        set_intersection(A.begin(), A.end(), B.begin(), B.end(), inserter(s2, s2.begin()));
        cout<<"A n B = "; print(s2);
        set_difference(A.begin(), A.end(), B.begin(), B.end(), inserter(s3, s3.begin()));
        cout<<"A - B = "; print(s3);
        set_difference(s1.begin(), s1.end(), A.begin(), A.end(), inserter(s4, s4.begin()));
        cout<<"SA = "; print(s4);
        set_difference(s1.begin(), s1.end(), B.begin(), B.end(), inserter(s5, s5.begin()));
        cout<<"SB = "; print(s5);
    }
    return 0;
}