试题 算法训练 P0705

资源限制

时间限制：1.0s 内存限制：256.0MB

　　输入两个整数集合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

        思路有很多种，这里我是利用了map，通过map储存该数在集合a和b的情况，map内部使用二叉树，所以可以保证输出有序。

#include<iostream>
#include<vector>
#include<string>
#include<math.h>
#include<fstream>
//#include<unordered_map>
#include<stack>
#include<queue>
#include<iomanip>
#include<algorithm>
#include<bitset>
#include<map>
using namespace std;

int main() {
    int n;
    cin >> n;
    //使用vector保存集合
    vector<int> nums1(n);
    //map表示数据个数
    map<int, int> mt;
    for (int i = 0; i < n; i++) {
        cin >> nums1[i];
        //a中数据使得加1
        mt[nums1[i]]++;
    }
    int m;
    cin >> m;
    vector<int> nums2(m);
    for (int i = 0; i < m; i++) {
        cin >> nums2[i];
        //b中减一
        mt[nums2[i]]--;
    }
    //使用auto会更加简洁，但是练习系统不支持
    map<int, int>::iterator it = mt.begin();
    
    for (it=mt.begin(); it != mt.end(); it++) {
        if(it->second==0)//为零表示二者中都有
            cout << it->first << " ";
    }
    cout << endl;
    for (it=mt.begin(); it != mt.end(); it++) {
        cout << it->first << " ";
    }
    cout << endl;
    for (it=mt.begin(); it != mt.end(); it++) {
        if (it->second == 1)//表示a有b没有
            cout << it->first << " ";
    }
    cout << endl;
    return 0;
}