看了书上的做法,学习了多种容器如何一起使用。。一开始完全不适应多种容器感觉要晕了。。看了好久才大概理解了下,,将其代码复述了出来。。
思路其实就是讲每个集合对应一个唯一ID,之后有了这个ID就可以找到这个集合,从而可以找出这个集合里面有几个元素。这里也利用了集合的一个特点,就是里面元素都是没有重复的。也就是说如果将一个集合里面已经有的元素插入到集合中时,这个集合其实是没有改变的。。
主要学习的就是这题的思考方式和思想。。(也需要日后的巩固)
附上代码:
#include<iostream>
#include<map>
#include<vector>
#include<string>
#include<set>
#include<stack>
#include<algorithm>
#define ALL(x) x.begin(),x.end()
#define INS(x) inserter(x,x.begin())
using namespace std;
typedef set<int> Set; //本题中的集合实际为集合的集合,(也就是集合里面还是一个个的集合,一开始还完全无法理解orz)
map<Set,int> IDcache; //将每个不同的集合对应唯一的一个ID
vector<Set> Setcache; //根据ID来取集合
int ID(Set x)
{
if(IDcache.count(x)) return IDcache[x];
Setcache.push_back(x);
IDcache[x]=Setcache.size()-1;
return IDcache[x];
}
int main ()
{
stack<int> s; //栈里存的是每个集合的ID 所以要用int
int n,t;
cin >> t;
while (t--)
{
cin >> n;
for(int i = 0;i < n;i++)
{
string a;
cin >> a;
if(a[0]=='P') s.push(ID(Set()));
else if(a[0]=='D') s.push(s.top());
else
{
Set x1 = Setcache[s.top()]; s.pop();
Set x2 = Setcache[s.top()]; s.pop();
Set x;
if(a[0] == 'U')
set_union(ALL(x1),ALL(x2),INS(x)); //并集
else if(a[0] == 'I')
set_intersection(ALL(x1),ALL(x2),INS(x)); //交集 (这两者都包含在algorithm里)
else if(a[0] == 'A')
{
x=x2;
x.insert(ID(x1));
}
s.push(ID(x));
}
cout << Setcache[s.top()].size() << endl; //用ID将栈顶元素取出来再看这个集合里面有多少的元素
}
cout << "***\n";
}
return 0;
}