3.7 PAT1016 乙级 python

部分A+B

Da,A,Db,B = input('').strip().split()
na = nb = ''

for i in range(len(Da)):
	if Da[i] == A:
		na += '1'
for i in range(len(Db)):
	if Db[i] == B:
		nb += '1'
if na == '' and nb == '':
	he = 0
elif nb == '':
	he = int(na)*int(A)
elif na == '':
	he = int(nb)*int(B)
else:
	he = int(na)*int(A) + int(nb)*int(B)
print(he)