[河南省ACM省赛-第五届] 试制品 （nyoj 542）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<set>
using namespace std;
#define N 402

vector<string>a
;//左式化合物集合
vector<string>b
;//右式化合物集合
vector<string>c;//存在的化合物
set<string>ans;//新生成化合物
bool found
;//标记化学式是否已经合成

int main() {
freopen("d:\\in.txt", "r", stdin);
int n, m;
while(cin>>n) {
memset(found, false, sizeof(found));
for(int i=0; i<n; i++){
a[i].clear();
b[i].clear();
}
c.clear();
ans.clear();
string expr, str;
for(int i=0; i<n; i++) {
cin>>expr;
int mid = expr.find("=", 0);
expr[mid] = '+';
expr += "+";
int len = expr.size();
for(int j = 0; j<mid+1; j++){
if(expr[j] != '+')
str.push_back(expr[j]);
else{
a[i].push_back(str);
str.clear();
}
}
for(int j = mid+1; j<len; j++){
if(expr[j] != '+')
str.push_back(expr[j]);
else{
b[i].push_back(str);
str.clear();
}
}
}
cin>>m;
for(int i=0; i<m; i++){
cin>>expr;
c.push_back(expr);
}
for(int i=0; i<n; i++){//排序方便查找
sort(a[i].begin(), a[i].end());
}
bool hasnext = true;
while(hasnext){
hasnext = false;
for(int k=0; k<n; k++){
if(found[k]) continue;
string first = a[k][0];
if(find(c.begin(), c.end(), first) != c.end()){//如果方程式的第一个元素存在
int size = a[k].size();
int i = 1;
for( ; i<size; i++){//在已经存在的化合物集合中查找化学式剩余的化合物
if(find(c.begin()+1, c.end(), a[k][i]) == c.end())
break;
}
if(i == size) {//如果左式完全匹配
found[k] = true;
hasnext = true;
for(vector<string>::iterator it = b[k].begin(); it != b[k].end(); it++){
if(find(c.begin(), c.end(), *it) == c.end()) {
c.push_back(*it);
ans.insert(*it);
}
}
}
}
}
}
cout<<ans.size()<<endl;
for(set<string>::iterator it = ans.begin(); it != ans.end(); it++)
cout<<*it<<endl;
}
return 0;
}