NYOJ1236 挑战密室

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

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

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

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

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

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

             输入

第一行： K，表示有K个化学方程式；

接下来有K行，每行为一个化学方程式

输出

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

样例输入

3

2C+O2=2CO2

NaOH+H2SO4=Na2SO4+2H2O

Ca2CO3+H2O=Ca2(OH)2+CO2

样例输出

0056

0142

0116

提示

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

来源

     

 上传者

      

//我用的是数组模拟，当然也可以用栈来做，和表达式求值是类似的
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
using namespace std;
int main()
{
    int t;
    scanf("%d",&t);
    getchar();
    while(t--)
    {
        int i,j,len,m,n;
        char a[1005];
        gets(a);
        len=strlen(a);
        int flag=0,ans=0,ants,tlag=1,tt,slag=0,antt=0;
        for(i=0;i<len;i++)
        {
            if(a[i-1]=='='&&a[i]>='2'&&a[i]<='9')//等号后面有数字的情况
                tlag=a[i]-'0';//表示等号后面有数字要乘的倍数，没有的话就是乘1，tlag初始值设为了1
            if(flag==1)//是等号后面需要判断的情况
            {
                if(a[i]=='(')//遇到左括号需要先把括号里的所有相对原子质量都加起来
                {
                    slag=1;
                }
                if(a[i]=='N'&&a[i+1]=='a')
                {
                    if(slag==1)//如若遇到左括号，以下判断都类似
                    antt+=23;//存放括号里的分子量，以下判断都类似
                    ants=23,ans+=ants,a[i+1]='1';//ans存放要求的分子量，ants存放当前加上的分子量，以下都类似
                }
                else if(a[i]=='N')
                {
                    if(slag==1)
                    antt+=14;
                    ants=14,ans+=ants;
                }
                else if(a[i]=='C'&&a[i+1]=='l')
                {
                    if(slag==1)
                    antt+=35;
                    ants=35,ans+=35,a[i+1]='1';
                }
                else if(a[i]=='C'&&a[i+1]=='a')
                {
                    if(slag==1)
                    antt+=40;
                    ants=40,ans+=40,a[i+1]='1';
                }
                else if(a[i]=='C')
                {
                    if(slag==1)
                    antt+=12;
                    ants=12,ans+=12;
                }
                else if(a[i]=='O')
                {
                    if(slag==1)
                    antt+=16;
                    ants=16,ans+=16;
                }
                else if(a[i]=='S')
                {
                    if(slag==1)
                    antt+=32;
                    ants=32,ans+=32;
                }
                else if(a[i]=='H')
                {
                    if(slag==1)
                    antt+=2;
                    ants=2,ans+=2;
                }
                else if(a[i]=='A'&&a[i+1]=='l')
                {
                    if(slag==1)
                    antt+=27;
                    ants=27,ans+=27,a[i+1]='1';
                }
                else if(a[i]=='Z'&&a[i+1]=='n')
                {
                    if(slag==1)
                    antt+=65;
                    ants=65,ans+=65,a[i+1]='1';
                }
                if(a[i]>='2'&&a[i]<='9'&&i!=tt+1)
                {
                    if(a[i-1]==')')
                    ans+=antt*(a[i]-'1');
                    else
                    ans+=(a[i]-'1')*ants;
                }
                if(a[i]=='+'||a[i]=='\n')//遇到'+'或字符串结束就停止计算
                    flag=0;
            }
            if(a[i]=='=')//遇到等号标记flag为1，标记tt是为了看后一个是不是数字
                flag=1,tt=i;
        }
        printf("%04d\n",ans*tlag);
    }
    return 0;
}