2020冬季PAT甲级 7-4 Chemical Equation (30 分)

明天就要考PAT了,从2月1号到今天大概40天了…
虽然本科课程有C和C++,2月1号之前我还是一个连hello world都不会写的零基础小孩…
今天写完最后一套真题,过去一直参考CSDN上大神的代码,今天想自己也发一篇…
这是我写的第一篇CSDN,有什么错误还请大家批评,我会改正的;


以下是正文
题目:7-4 Chemical Equation (30 分)
2020冬季PAT甲级 7-4 Chemical Equation (30 分)_第1张图片

#include 
#include 
#include 
#include 
#pragma warning(disable:4996)//这是因为VS不能直接用scanf
using namespace std;

int N, M, K, r;
vector<int> R, P, rea, path;//R是反应物,P是要求生成的物质,rea是个工具数组...,path就是答案啦;
vector <vector<int> > react[110];//react里存放所有的反应,react[i][j][k]的意思是反应物i的第j优先级的反应式里面有k这种物质;上面的path里面存放每个i的j;
bool use[110] = {
      false };//判断这个反应物有没有用过;

bool dfs(int i) {
     
	if (i == M) {
     //只要走到最后一步就可以输出了,因为优先级已经排好了,走到最后就是最优先的;
		for (int j = 0; j < M; j++) {
     
			for (int t = 0; t < react[P[j]][path[j]].size(); t++) {
     
				if (t != 0) printf(" + ");
				printf("%02d", react[P[j]][path[j]][t]);
			}
			printf(" -> %02d\n", P[j]);
		}
		return 1;
	}
	for (int t = 0; t < react[P[i]].size(); t++) {
     
		int j;
		for (j = 0; j < react[P[i]][t].size() && use[react[P[i]][t][j]] == 1; j++);//检查这个方案的反应物是不是都没被用
		if (j == react[P[i]][t].size()) {
     
			for (j = 0; j < react[P[i]][t].size(); j++)//选了这个方案,把反应物用掉!
				use[react[P[i]][t][j]] = 0;
			path.push_back(t);
			if (dfs(i + 1))//看看下一个物质选哪个方案
				return 1;
			else {
     
				for (j = 0; j < react[P[i]][t].size(); j++)//这个方案不行,把反应物还回来!
					use[react[P[i]][t][j]] = 1;
				path.pop_back();
			}
		}
	}
	return 0;//这个物质的所有方案都不行,原因是之前的物质把反应物用掉了,要改之前的方案;
}

bool cmp(vector<int>a, vector<int>b) {
     
	for (int i = 0;; i++) {
     
		if (a[i] != b[i])
			return a[i] < b[i];
	}
}
int main() {
     
	scanf("%d", &N);
	string s;
	for (int t = 0; t < N; t++) {
     
		scanf("%d", &r);
		R.push_back(r);
		use[r] = 1;
	}
	scanf("%d", &M);
	for (int t = 0; t < M; t++) {
     
		scanf("%d", &r);
		P.push_back(r);
		if (find(R.begin(), R.end(), r) != R.end()) {
     //生成物同时也是反应物,加一个反应方案;
			rea.push_back(r);
			react[r].push_back(rea);
			rea.clear();
		}
	}
	scanf("%d\n", &K);
	while (K--) {
     
		getline(cin, s);
		int flag = 0;
		sscanf(s.c_str(), "%d", &r);
		rea.push_back(r);
		s = s.substr(3);
		while (s[0] != '-') {
     
			sscanf(s.c_str(), "+ %d", &r);
			s = s.substr(5);
			rea.push_back(r);
		}
		sscanf(s.c_str(), "-> %d", &r);//找到最后的生成物
		react[r].push_back(rea);//储存方案;
		rea.clear();
	}
	for (int i = 0; i < M; i++) {
     //给反应方案排序;
		sort(react[P[i]].begin(), react[P[i]].end(), cmp);
	}
	dfs(0);
	return 0;
}

我学的不多,懂得真的挺少的所以,就这些啦;
希望大家都能考个好成绩!

你可能感兴趣的:(c++,dfs)