一、题目
[题目描述]
化学方程式,也称为化学反应方程式,是用化学式表示化学反应的式子。给出- -组
化学方程式,请你编写程序判断每个方程式是否配平(也就是方程式中等号左右两边的
元素种类和对应的原子个数是否相同)。
本题给出的化学方程式由大小写字母、数字和符号(包括等号=、加号+、左圆括
号(和右圆括号))组成,不会出现其他字符(包括空白字符,如空格、制表符等)。化
学方程式的格式与化学课本中的形式基本相同(化学式中表示元素原子个数的下标用.
正常文本,如H2O写成H20),用自然语言描述如下:
●化学方程式由左右两个表达式组成,中间用一-个等号=连接,如2H2+02=2H20;
●表达式由若干部分组成,每部分由系数和化学式构成,部分之间用加号+连接,
如2H2+02、2H20;
●系数是整数或空串,如为空串表示系数为1;
●整数由一个或多个数字构成; .
●化学式由若干部分组成,每部分由项和系数构成,部分之间直接连接,如H20、
C02、Ca(0H)2、 Ba3(P04)2;
●项是元素或用左右圆括号括起来的化学式,如H、Ca、(OH)、 (P04);
●元素可以是一个大写字母,也可以是一个大写字母跟着一一个小写字母,如H、0、
Ca。
二、分析
有一天早上突然起来,发现自己想漏了一种情况…遂重新写了一份,丢进OJ测试后是满分。
写CSP的大佬们大多是用的C/C++,百度了一下,python的满分代码很少,只找到了一份281ms的代码,还有一些代码跟我之前CCF CSP 201912-3 化学方程式 【Python版】这个版本一样,其实不是超时,是想漏了——官方给的测试点并不全面,需要自己脑中穷举一波。
这道题我能想到的办法是用栈来递归下降,在python性能有限的情况下,这样写出来只需要125ms,我个人还是比较满意。虽然可能还可以继续优化,但我还是先贴出来供大家参考,欢迎大家交流指正。
三、代码部分
# -*- coding: utf-8 -*-
"""
Spyder Editor
This is a temporary script file.
"""
'''
11
Cu+As=Cs+Au
2H2+O2=2H2O
H2+Cl2=2NaCl
H2+Cl2=2HCl
CH4+2O2=CO2+2H2O
CaCl2+2AgNO3=Ca(NO3)2+2AgCl
3Ba(OH)2+2H3PO4=6H2O+Ba3(PO4)2
3Ba(OH)2+2H3PO4=Ba3(PO4)2+6H2O
4Zn+10HNO3=4Zn(NO3)2+NH4NO3+3H2O
4Au+8NaCN+2H2O+O2=4Na(Au(CN)2)+4NaOH
H2+O2=H2O
'''
class E:
def __init__(self,in_str):
def getNum(i,instr):
temp_num=''
mylen=len(instr)
while i<mylen:
if instr[i].isdigit():
temp_num+=(instr[i])
i+=1
else:
break
return i,int(temp_num)
def apElem(indict,name,num=1):
if name in indict:
indict[name]+=num
else:
indict[name]=num
def disUpper(indict,instr,i):
temps=''
sub=1
mylen=len(instr)
temps+=instr[i]
i+=1
if i>=mylen:
apElem(indict,temps,sub)
return i
else:
if instr[i].islower():
temps+=instr[i]
i+=1
if i>=mylen:
apElem(indict,temps,sub)
return i
elif instr[i].isdigit():
i,sub=getNum(i,instr)
apElem(indict,temps,sub)
return i
else:
apElem(indict,temps,sub)
return i
elif instr[i].isdigit():
i,sub=getNum(i,instr)
apElem(indict,temps,sub)
return i
else:
apElem(indict,temps,sub)
return i
def mergeDict(d1,d2):
for i in d2.keys():
if i in d1:
d1[i]+=d2[i]
else:
d1[i]=d2[i]
def updateCoef(dc,coef):
for i in dc.keys():
dc[i]*=coef
def disBrackets(indict,instr,i):
dt={}
listb=[]
listb.append(i+1)
i+=1
mylen=len(instr)
while i<mylen:
if instr[i]=='(':
listb.append(i+1)
elif instr[i]==')':
l=listb.pop()
if(not listb):
r=i
ts=instr[l:r]
disForm(dt,ts)
i+=1
break
i+=1
if(i<mylen and instr[i].isdigit()):
i,tempsub=getNum(i,instr)
updateCoef(dt,tempsub)
mergeDict(indict,dt)
return i
def disForm(indict,instr):
fdict={}
mycoef=-1
i = 0
mylen=len(instr)
while i < mylen:
if i==0 and instr[i].isdigit():
i,mycoef=getNum(i,instr)
elif instr[i].isupper():
i=disUpper(fdict,instr,i)
elif instr[i]=='(':
i=disBrackets(fdict,instr,i)
else:
i+=1
if mycoef!=-1:
updateCoef(fdict,mycoef)
mergeDict(indict,fdict)
instr=in_str.split('=')
instr_l=instr[0].split('+')
instr_r = instr[1].split('+')
dictL={}
dictR={}
for i in instr_l:
disForm(dictL,i)
for j in instr_r:
disForm(dictR,j)
if dictL==dictR:
print('Y')
else:
print('N')
n=int(input())
for i in range(n):
e=E(input())
请点个赞吧,拜托了TAT