TOJ1743 集合运算 Set

描述

给定两个集合A和B的所有元素，计算它们的交、并、差集。

输入

输入数据有多组，第一行为数据的组数T，接下来有2T行，每组数据占2行，每行有若干个整数，第一行的所有整数构成集合A，第二行的所有整数构成集合B，分别用空格分隔。A和B最多分别不超过100个元素。

输出

输出A、B的交、并、差集中的所有元素（以递增顺序）。每个集合占一行，数据元素之间用空格分隔。

（输入输出显示不出来... 用图片凑合一下吧...

这道题充分让我感受到了C++的可爱以及Set的强大...

用输入流输入很舒服，Set存好之后用Set本身的函数就可以得到交、并、差集。

然后多组数据记得清空Set。

#include<stdio.h>
#include<string.h>
#include<string>
#include<math.h>
#include<sstream>
#include<set>
#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
	int t,i,j,a,b,k;
	set<int> s1,s2,s3,s4,s5;
	string str1,str2;
	cin>>t;
	getchar();
	while(t--)
	{
		s1.clear();
	 	s2.clear();
	 	s3.clear();
	 	s4.clear();
	 	s5.clear();
		//输入 
		getline(cin,str1);
		stringstream ss(str1);		
		while(ss>>a)
		{
			s1.insert(a);
		}
		getline(cin,str2);
		stringstream ss2(str2);
		while (ss2>>b)
		{
			s2.insert(b);
		}
		//计算 
		set_intersection(s1.begin(),s1.end(),s2.begin(),s2.end(),inserter(s3,s3.begin()));
		set_union(s1.begin(),s1.end(),s2.begin(),s2.end(),inserter(s4,s4.begin()));
		set_difference(s1.begin(),s1.end(),s2.begin(),s2.end(),inserter(s5,s5.begin()));
		//输出 
		set<int>::const_iterator it;
		k=0;
		for(it=s3.begin();it!=s3.end();it++)
		{
			if(k==1)
			cout<<" ";
			cout<<*it;
			k=1;
		}
		cout<<endl;
		k=0;
		for(it=s4.begin();it!=s4.end();it++)
		{
			if(k==1)
			cout<<" ";
			cout<<*it;
			k=1;
		}
		cout<<endl;
		k=0;
		for(it=s5.begin();it!=s5.end();it++)
		{
			if(k==1)
			cout<<" ";
			cout<<*it;
			k=1;
		}
		cout<<endl;
	}
}