PTA 1016 部分A+B (15分) (python 3)

正整数 A 的“D​A​​(为 1 位整数)部分”定义为由 A 中所有 D​A​​ 组成的新整数 P​A​​。例如:给定 A=3862767,D​A​​=6,则 A 的“6 部分”P​A​​ 是 66,因为 A 中有 2 个 6。
现给定 A、D​A​​、B、D​B​​,请编写程序计算 P​A​​+P​B​​。
输入格式:
输入在一行中依次给出 A、D​A​​、B、D​B​​,中间以空格分隔,其中 0<A,B<10​10​​。
输出格式:
在一行中输出 P​A​​+P​B​​ 的值。
输入样例 1:

3862767 6 13530293 3

输出样例 1:

399

输入样例 2:

3862767 1 13530293 8

输出样例 2:

0

代码片

A,D1,B,D2=input().split()
n1=A.count(D1)
n2=B.count(D2)
if n1!=0 and n2==0:
    Pa=D1*n1
    Pb='0'
elif n1==0 and n2!=0:
    Pa='0'
    Pb=D2*n2
elif n1==0 and n2==0:
    Pa='0'
    Pb='0'
else:
     Pa=D1*n1
     Pb=D2*n2   
P=int(Pa)+int(Pb)
print(P)
发布了9 篇原创文章 · 获赞 0 · 访问量 182

猜你喜欢

转载自blog.csdn.net/qq_46258819/article/details/104156774