试题算法训练集合运算

╰+攻爆jí腚メ 2024-04-01 02:23 80阅读 0赞

## 试题 算法训练 集合运算 ##

资源限制  
内存限制：512.0MB C/C++时间限制：1.0s Java时间限制：3.0s Python时间限制：5.0s  
问题描述  
　　给出两个整数集合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

**提交代码**：

#include<bits/stdc++.h>
    using namespace std;
    int n, m, a;
    map<int, int> mp;
    int main(){
        
    	cin >> n;
    	for(int i = 0; i < n; i++) {
        
    		cin >> a; mp[a] = 1;  // 等于1代表只有集合a有这个数
    	}
    	cin >> m;
    	for(int i = 0; i < m; i++) {
        
    		cin >> a;
    		if(mp[a]) mp[a] = 3;  // 等于3代表集合a和集合b都有这个数
    		else mp[a] = 2;       // 等于2代表只有集合b有这个数
    	}
    	for(map<int, int>::iterator it = mp.begin(); it != mp.end(); it++){
        
    		if(it->second == 3) cout << it->first << " ";
    	} 
    	cout << endl;
    	for(map<int, int>::iterator it = mp.begin(); it != mp.end(); it++){
        
    		cout << it->first << " ";
    	} 
    	cout << endl;
    	for(map<int, int>::iterator it = mp.begin(); it != mp.end(); it++){
        
    		if(it->second == 1) cout << it->first << " ";
    	} 
    }