紫书的STL介绍,比较抽象,教我们怎么表示一些抽象的集合并进行一些简单的操作,其中Set_Set_ID和vector Set_Set;是关键,然后关于set_union,set_intersection在下面链接中有介绍
#include <iostream>
#include <set>
#include <map>
#include <stack>
#include <vector>
#include <algorithm>
#include <string>
#define ALL(x) x.begin(),x.end()
#define INS(x) inserter(x,x.begin())
using namespace std;
typedef set<int> Set_Int; //代表了一个集合,用不同的ID号表示不同的集合
map<Set_Int, int> Set_Set_ID;//用来表示集合的集合中不同集合和该集合ID的映射
vector<Set_Int> Set_Set;//表示集合的集合 ,用来存储集合
stack<int> my_stack; //集合栈,存储集合对应的ID
int GET_ID(Set_Int x)
{
if (Set_Set_ID.count(x)) return Set_Set_ID[x];
Set_Set.push_back(x);//如果未在集合中,则添加一个集合
Set_Set_ID[x] = Set_Set.size() - 1; //这里这样赋值保证Set_Set中的元素的ID和my_stack栈中的顺序一致,方便取出
//所以这里Set_Set不能用set类型,而要用vector,因为set类型默认是有从小到大
//排序的,我们并不能保证它按照my_stack中的顺序来
return Set_Set_ID[x];
}
int main(void)
{
int n;
cin >> n;
while (n--)
{
while (!my_stack.empty())my_stack.pop();
while (!Set_Set.empty())Set_Set.pop_back();
while (!Set_Set_ID.empty())Set_Set_ID.clear();
int m;
cin >> m;
for (int i = 0; i<m; i++)
{
string option;
cin >> option;
if (option[0] == 'P')
my_stack.push(GET_ID(Set_Int()));//插入空集
else if (option[0] == 'D')
my_stack.push(my_stack.top());
else
{
Set_Int set1 = Set_Set[my_stack.top()]; my_stack.pop();
Set_Int set2 = Set_Set[my_stack.top()]; my_stack.pop();
Set_Int result_set;
if (option[0] == 'U')
set_union(ALL(set1), ALL(set2), INS(result_set));
if (option[0] == 'I')
set_intersection(ALL(set1), ALL(set2), INS(result_set));
if (option[0] == 'A') { result_set = set2; result_set.insert(GET_ID(set1)); }
my_stack.push(GET_ID(result_set));
}
cout << Set_Set[my_stack.top()].size() << endl;
}
cout << "***" << endl;
}
return 0;
}