2020-12-05 冬季pat甲级 D Chemical Equation

感觉这次pat考试最麻烦的题就是这个第四题了，所以决定留个代码记录下。

---

题目大意：
有n个反应物 ，编号分别为a1,a2……an（不一定是从1-n)

再给出m个产物：b1, b2……bm

现有k个反应式

每一个反应式 形如：a1 + a2 …… + aj -> bi， bi为产物；

问 ： 用所给出的n个反应物 怎样生成这m个产物 （每个反应物只能用一次，保证存在解）

输出要求： 对于每个产物 bi， 输出其反应式：
如：
a1 + a2 …… + an -> bi;

（若某反应物自己本身就是产物也是可以的， 直接输出 a1->a1， 如果存在多种可能选择编号最小的)

---

Input：

8 09 05 03 04 02 01 16 10
3 08 03 04
6
03 + 09 -> 08
02 + 08 -> 04
02 + 04 -> 03
01 + 05 -> 03
01 + 09 + 16 -> 03
02 + 03 + 05 -> 08

Output：

02 + 03 + 05 -> 08
01 + 09 + 16 -> 03
04 -> 04

---

刚看到题目第一反应就是状态压缩，因为n好像是小于等于20的，就想着直接暴力枚举了

for(int i  = 1; i < (1 << n); i++){
	//判断当前状态是否满足生成m个产物；
}

但好像这个判断不太好实现，有可能会超时

后面想了下决定用离散化（这步可以省略，因为好像题目默认给出的编号是不会大于20的）加 状态压缩处理反应表达式， 再DFS。

把反应物排序后进行标号，再对反应式进行处理，把每个反应所需要的反应物用一个二进制数来表示，并存在相应产物的数组里。最后进行DFS，通过位运算 & 判断所选反应物是否出现交集。

细节处理：离散化操作时，从大到小赋值。后续对每种产物可能的反应物方案排序从大到小排序。这样就可以保证DFS第一次找到的可行方案即为题干要求的最小序列

---

AC码

#include <bits/stdc++.h>
#include <unordered_map>
using namespace std;

set<int>s;
map<int, int>dic;
map<int, int>dic_;
map<int, vector<int>> v;
int a[15];
int n, m;
vector<vector<int>> vv;


bool DFS(int cur, int index) {
	if (index == m)
		return 1;
	bool ret = 0;
	for (int i = 0; i < v[a[index]].size(); i++) {
		int ans = v[a[index]][i];
		if ((cur & ans) == 0) {
			if (DFS(cur + ans, index + 1)){
				vector<int>now;
				for (int k = n; k >= 0; k--)
					if (ans & (1 << k))
						now.push_back(dic_[k]);
				now.push_back(a[index]);
				vv.push_back(now);
				return 1;
			}
		}
	}
	return 0;
}

void Print() {
	for (int i = m - 1; i >= 0; i--) {
		if (vv[i].size() == 0)
			continue;
		for (int j = 0; j < vv[i].size(); j++) {
			if (j != 0 && j != vv[i].size() - 1)
				cout << " + ";
			if (j == vv[i].size() - 1)
				cout << " -> ";
			if (vv[i][j] < 10)
				cout << 0;
			cout << vv[i][j];
		}
		cout << endl;
	}
}

int main() {
	cin >> n;
	for (int i = 0; i < n; i++) {
		int k;
		cin >> k;
		s.insert(k);
	}
	int index = n;
	for (auto it : s) {
		dic[it] = index;
		dic_[index--] = it;
	}
	cin >> m;
	for (int i = 0; i < m; i++) {
		cin >> a[i];
		if (dic.find(a[i]) != dic.end())
			v[a[i]].push_back((1 << dic[a[i]]));
	}
	int k;
	cin >> k;
	while (k--) {
		string ss;
		getline(cin, ss);
		if (ss == "") {
			k++;
			continue;
		}
		int cur = 0, jud = 0, add = 1, ans = 0;
		for (int i = 0; i < ss.size(); i++) {
			if (ss[i] >= '0' && ss[i] <= '9') {
				jud = 1;
				cur = cur * 10 + ss[i] - '0';
			}
			else if (ss[i] == ' ' && jud == 1) {
				jud = 0;
				if (dic.find(cur) == dic.end()) {
					add = 0;
					break;
				}
				ans += (1 << dic[cur]);
				cur = 0;
			}
		}
		if (add)
			v[cur].push_back(ans);
	}
	for (auto& it : v)
		sort(it.second.begin(), it.second.end(), greater<int>());//排序使获得的第一个方案就是最小的
	DFS(0, 0);
	Print();
}