CCF 化学方程式 c++(100分)

将CCF22次考试的t3都做了一遍，发现t3的模拟题挺爱出这种需要递归处理嵌套的题目，这道化学方程式，还有json查询，以及元素选择器，都是神似的题目。

处理字符串的话，字符串流是极好的方法，如本题，只要通过一次循环将化学方程式中的'+'全部替换为' ',然后通过’=‘的位置将化学方程式分成左右两部分。

每部分我们通过空格可以分为若干个化合物表达式，我们处理的基本单位就是一个化合物的表达式。

题目中涉及数字的地方很多，我们可以写个函数来获取数字。

同时注意，我们通过传引用的方式，共用一个索引。

1.如果当前是大写字母，就查询下一个字符是否是小写字符，如果是的话，加起来就是元素的名称。然后查询元素名称之后是否有数字。

2.如果当前是'('的话，说明我们要递归处理。构造一个临时map变量作为参数，然后等递归调用完，我们让本次递归的map和临时变量map相加。

3.如果是')'的话，说明本次递归处理结束，查寻一下')'后面是否有数字，如果有数字，则对各原子的数目乘上该数字。然后return

源代码如下

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;

typedef pair<LL,LL> pa;

int n;

int getnum(string &s,int &k)

{

    //从k开始寻找数字

    int sz=s.size();

    string w;

    while(k<sz&&isdigit(s[k]))

        w+=s[k++];

    if(w.empty())return 1;

    return stoi(w);

    //返回的k是第一个不是数字的位置

}

void solve(string &s,map<string,int>&m,int &k)

{

    int w=getnum(s,k);//化学式前的系数

    int sz=s.size();

    string str;//名称

    while(k<sz){

        if(isupper(s[k])){//得到元素的名称

            str=s[k];

            if(islower(s[k+1]))str+=s[++k];

            k++;//过滤字母

            int tmpw=getnum(s,k);//获得该元素的数量

            m[str]+=tmpw;

         //   cout<<" str "<<str<<" num "<<tmpw<<endl;

            str.clear();//清空

        }

        else{//只能是括号了

            if(s[k]=='('){

                map<string,int> tmpm;

                solve(s,tmpm,++k);//从k的下一个数开始

                for(auto t:tmpm){

                    m[t.first]+=tmpm[t.first];

                }

            }

            else if(s[k]==')'){

                int tmpw=getnum(s,++k);

                for(auto t:m)m[t.first]*=tmpw;

                return ;

            }

        } 

    }

    for(auto t:m)//这是处理的化学方程式前的系数

        m[t.first]*=w;

}

int main()

{

  //  freopen("in.txt","r",stdin);

  //  freopen("out.txt","w",stdout);

    std::ios::sync_with_stdio(false);

    cin>>n;

    string line;

    string str;

    getline(cin,line);

    while(n--){

        getline(cin,line);

        int index=0;

        while((index=line.find('+',index))!=line.npos){

            line[index]=' ';

            index++;

        }

        index=line.find('=');//分成两块

        vector<string> v1,v2;

        istringstream s1(line.substr(0,index)),s2(line.substr(index+1));

        while(s1>>str)v1.push_back(str);

        while(s2>>str)v2.push_back(str);

        map<string,int> m1,m2;

        int k;

        for(auto t:v1)

            {

            int k=0;

            map<string,int> tmpm;

            solve(t,tmpm,k);

            for(auto x:tmpm)

                m1[x.first]+=tmpm[x.first];

            }

        for(auto t:v2)

            {

            int k=0;

            map<string,int> tmpm;

            solve(t,tmpm,k);

            for(auto x:tmpm)

                m2[x.first]+=tmpm[x.first];

            }

        if(m1==m2)

            cout<<"Y"<<endl;

        else cout<<"N"<<endl;

    }

}