CCF201912-3 化学方程式

Adam_Eden_

已于 2023-03-05 18:38:20 修改

阅读量87

点赞数

文章标签： python 开发语言算法

于 2023-03-05 18:37:01 首次发布

本文链接：https://blog.csdn.net/qq_53220115/article/details/129349271

版权

CCF201912-3 化学方程式

题目连接

化学方程式

100分代码

# 100
def solve(expr: str) -> dict:
    d = {}

    def add(k: str, v: int):
        if k not in d:
            d[k] = v
        else:
            d[k] += v

    def deal(formula: str, coef: int):
        num = 0
        element = ''
        i = 0
        while i < len(formula):
            if 'A' <= formula[i] <= 'Z':
                if element != '':
                    add(element, coef * max(1, num))
                    element = ''
                    num = 0
                element = formula[i]
                i += 1
            elif 'a' <= formula[i] <= 'z':
                element = element + formula[i]
                i += 1
            elif '0' <= formula[i] <= '9':
                num = num * 10 + int(formula[i])
                i += 1
            elif formula[i] == '(':
                if element != '':
                    add(element, coef * max(1, num))
                    element = ''
                    num = 0
                j = i
                j += 1
                matching = -1
                while matching != 0:
                    if formula[j] == '(':
                        matching -= 1
                    if formula[j] == ')':
                        matching += 1
                    j += 1
                t = j
                while t < len(formula) and '0' <= formula[t] <= '9':
                    num = num * 10 + int(formula[t])
                    t += 1
                deal(formula[i + 1:j - 1], max(1, num) * coef)
                num = 0
                i = t
        if element != '':
            add(element, coef * max(1, num))

    for formula in expr.split('+'):
        num = 0
        i = 0
        while i < len(formula):
            if not '0' <= formula[i] <= '9':
                deal(formula[i:], max(num, 1))
                break
            else:
                num = num * 10 + int(formula[i])
                i += 1
    return d


def main():
    n = int(input())
    for i in range(n):
        equation = input()
        left, right = equation.split('=')[0], equation.split('=')[1]
        d1, d2 = solve(left), solve(right)
        if len(d1) != len(d2):
            print('N')
        else:
            flag = True
            for k, v in d1.items():
                if k not in d2 or d2[k] != v:
                    print('N')
                    flag = False
                    break
            if flag:
                print('Y')


if __name__ == '__main__':
    main()