NYOJ_1236_挑战密室

/*

挑战密室

时间限制：1000 ms  |  内存限制：65535 KB

难度：4

描述

R组织的特工Dr.Kong 为了寻找丢失的超体元素，不幸陷入WTO密室。Dr.Kong必须尽快找到解锁密码逃离，否则几分钟之后，WTO密室即将爆炸。

 

Dr. Kong发现密室的墙上写了许多化学方程式中。化学方程式，也称为化学反应方程式，是用化学式表示物质化学反应的式子。化学方程式反映的是客观事实。因此书写化学方程式要遵守两个原则：一是必须以客观事实为基础；二是要遵守质量守恒定律。

化学方程式不仅表明了反应物、生成物和反应条件。同时，化学计量数代表了各反应物、生成物物质的量关系，通过相对分子质量或相对原子质量还可以表示各物质之间的质量关系，即各物质之间的质量比。对于气体反应物、生成物，还可以直接通过化学计量数得出体积比。例如：2NaOH+H₂SO₄=Na₂SO₄+2H₂O

 

经过多次试探、推理，Dr. Kong发现密码是4位数字，就隐藏在化学方程式等号后的第一个分子中，其分子量就可能是密码（若分子量不足4位，前面加0）。

好在Dr.Kong还记得墙上各化学方程式用到的化学元素的原子量如下：

 

N	C	O	Cl	S	H	Al	Ca	Zn	Na
14	12	16	35	32	2	27	40	65	23

 

你能帮Dr. Kong尽快找到密码吗？

输入

第一行： K，表示有K个化学方程式；
接下来有K行，每行为一个化学方程式

输出

对于每个化学方程式输出一行：即密码。

样例输入

32C+O2=2CO2NaOH+H2SO4=Na2SO4+2H2OCa2CO3+H2O=Ca2(OH)2+CO2

样例输出

005601420116

提示

2≤K≤8 ，化学方程式的长度不超过50, 所有原子,分子的数量不超过9.小括号最多一层.

*/

 
#include <stdio.h>
#include <string.h>
#include <stack>
using namespace std;	

char letter[12][5] = {"N","C","O","Cl","S","H","Al","Ca","Zn","Na"};
int figure[12] = {14,12,16,35,32,2,27,40,65,23};
char s[55];

struct Node
{
	char ch[5];
}N;
int num,ans,res;
stack<Node> S;

int solve(char *c)
{
	int i;
	for(i=0;i<10;i++)
	{
		if(strcmp(c,letter[i])==0)
		{
			return figure[i];
		}
	}
}

int find(int i)
{
	int sum = 0;	
	if(s[i]>='A' && s[i]<='Z' && s[i+1]>='a' && s[i+1]<='z')
	{
		N.ch[0] = s[i];
		N.ch[1] = s[i+1];
		N.ch[2] = '\0';
		S.push(N);
	}
	else if(s[i]>='A' && s[i]<='Z')
	{
		N.ch[0] = s[i];
		N.ch[1] = '\0';
		S.push(N);
	}
	else if(s[i]>'1' && s[i]<='9')
	{
		Node p = S.top();
		sum += solve(p.ch) * (s[i]-'0');
		S.pop();
	}
	return sum;
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		getchar();
		int i=0,j;
		ans = 0, res = 0 ,num = 1;
		scanf("%s",s);
		int len = strlen(s);
		while(s[i++]!='=');
		if(s[i]>'1' && s[i]<='9')
		{
			num = s[i++] - '0';
		}
		while(s[i]!='+' && i<len)
		{
			ans += find(i);
			i++;
			if(s[i] == '(')
			{
				N.ch[0] = '(';
				N.ch[1] = '\0';
				S.push(N);
				while(s[i]!=')')
				{
					res += find(i);
					i++;	
				}
				Node p = S.top();
				S.pop();
				while(strcmp(p.ch,"(")!=0)
				{
					res += solve(p.ch);
					p = S.top();
					S.pop();
				}
				if(s[++i]>'1' && s[i]<='9')
				{
					res *= (s[i++] - '0');
				}
			}
		}
		while(!S.empty())
		{
			Node p = S.top();
			ans += solve(p.ch);
			S.pop();
		}
		int res_s = num * (ans + res);
		if(res_s<10)
		{
			printf("000%d\n",res_s);
		}
		else if(res_s<100)
		{
			printf("00%d\n",res_s);
		}
		else if(res_s<1000)
		{
			printf("0%d\n",res_s);
		}
		else
		{
			printf("%d\n",res_s);
		}
	}
	return 0;
}