CCF201912-3 化学方程式（100分）文本处理

Posted 2021-09-10 海岛Blog

试题编号： 201912-3
试题名称： 化学方程式
时间限制： 1.0s
内存限制： 512.0MB

问题链接：CCF201912-3 化学方程式
问题简述：（略）
问题分析：文本处理问题，按字符串来处理即可，顺序处理，递归都不需要。
程序说明：（略）
参考链接：（略）
题记：（略）

100分的C++语言程序如下：

/* CCF201912-3 化学方程式 */

#include <bits/stdc++.h>

using namespace std;

struct Elem { //元素
    string name; //名称
    int num; //个数
    Elem(string name1, int num1): name(name1), num(num1){}
};

int myatoi(string s, int &pos)
{
    int num = 0;
    while (isdigit(s[pos])) num = num * 10 + s[pos++] - '0';
    return num;
}

void calc(string &s, map<string, int> &mp)
{
    stringstream ss(s);
    string t;

    while(getline(ss, t, '+')){ // 获取化学式t
        vector<Elem> a;
        int factor = 1;
        int i = 0;

        if(isdigit(t[i])) factor = myatoi(t, i);

        while (i < (int)t.size()) {
            if (isdigit(t[i])) {
                int num = myatoi(t, i);
                if (a[a.size() - 1].name == ")") {
                    int j = a.size() - 1;
                    a[j].name = "*";
                    while (a[--j].name != "(") a[j].num *= num;
                    a[j].name = "*";
                } else
                    a[a.size() - 1].num *= num;
            } else if (t[i] == '(') {
                a.push_back(Elem("(", 0));
                i++;
            } else if (t[i] == ')') {
                a.push_back(Elem(")", 0));
                if (i + 1 == (int)t.size() || !isdigit(t[i + 1]))
                    t.insert(i + 1, "1");
                i++;
            } else if (isupper(t[i])) {
                string name;
                name += t[i++];
                if (islower(t[i])) name += t[i++];
                a.push_back(Elem(name, 1));
            }
        }

        for (int i = 0; i < (int)a.size(); i++)
            if (a[i].name != "*") mp[a[i].name] += a[i].num * factor;
    }
}

bool judge(map<string, int> &left, map<string, int> &right)
{
    if (left.size() != right.size()) return false;
    for (map<string, int>::iterator iter = left.begin(); iter != left.end(); iter++)
        if (right[iter->first] != iter->second) return false;
    return true;
}

int main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int n;
    cin >> n;
    for (int i = 1; i <= n; i++) {
        string s, l, r;
        cin >> s;
        stringstream ss(s);
        getline(ss, l, '=');
        getline(ss, r);

        map<string, int> left, right;
        calc(l, left);
        calc(r, right);

        cout << (judge(left, right) ? 'Y' : 'N') << endl;
    }

    return 0;
}