POJ 2269 Friends（模拟）

Description

给出集合交并差运算表达式，输出结果

+:并运算 -:差运算 *:交运算

Input

多组用例，每组用例一个表达式，以文件尾结束输入

Output

对于每组用例，输出表达式运算结果

Sample Input

{ABC}

{ABC}+{DEFG}+{Z}+{}

{ABE}*{ABCD}

{ABCD}-{CZ}

{ABC}+{CDE}*{CEZ}

({ABC}+{CDE})*{CEZ}

Sample Output

{ABC}

{ABCDEFGZ}

{AB}

{ABD}

{ABCE}

{CE}

Solution

模拟，先对表达式进行处理，然后运用set的交并差函数进行运算

Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<string>
#include<set>
#include<stack>
#include<iterator>
using namespace std;
#define maxn 300
char temp[maxn];
int value[128];
int stacki;
string c,s;
void adjust()//对输入的表达式进行处理
{
s.clear();
stacki=0;
for(int i=0;i!=c.size();i++)
if(c[i]=='(')
temp[stacki++]=-1;
else if(c[i]==')')
{
while(temp[stacki-1]!=-1)
s.push_back(temp[--stacki]);
stacki--;
}
else if(c[i]=='{')
{
for(;c[i]!='}';i++)
s.push_back(c[i]);
s.push_back(c[i]);
}
else
{
while(stacki!=0&&temp[stacki-1]!=-1&&value[temp[stacki-1]]>=value[c[i]])
s.push_back(temp[--stacki]);
temp[stacki++]=c[i];
}
while(stacki!=0)
s.push_back(temp[--stacki]);
}
set<char>solve()
{
set<char>stack[maxn];//存储子表达式
set<char>temp;
stacki=0;
for(int i=0;i!=s.size();i++)
if(s[i]=='{')//将子表达式压入栈中
{
i++;
stack[stacki].clear();
for(;s[i]!='}';i++)
stack[stacki].insert(s[i]);
stacki++;
}
else if(s[i]=='+')//并运算
{
--stacki;
temp.clear();
set_union(stack[stacki-1].begin(),stack[stacki-1].end(),
stack[stacki].begin(),stack[stacki].end(),
inserter(temp,temp.begin()));
stack[stacki-1]=temp;
}
else if(s[i]=='-')//差运算
{
--stacki;
temp.clear();
set_difference(stack[stacki-1].begin(),stack[stacki-1].end(),
stack[stacki].begin(),stack[stacki].end(),
inserter(temp,temp.begin()));
stack[stacki-1]=temp;
}
else//交运算
{
--stacki;
temp.clear();
set_intersection(stack[stacki-1].begin(),stack[stacki-1].end(),
stack[stacki].begin(),stack[stacki].end(),
inserter(temp,temp.begin()));
stack[stacki-1]=temp;
}
return stack[0];
}
int main()
{
value['+']=value['-']=0;//+和-的优先级较低
value['*']=1;//*的优先级较高
while(cin>>c)
{
adjust();
set<char>ans=solve();
printf("{");
for(set<char>::iterator i=ans.begin();i!=ans.end();i++)
printf("%c",*i);
printf("}\n");
}
return 0;
}