PAT 1016 部分A+B python

1016 部分A+B（15 分）

正整数 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：

扫描二维码关注公众号，回复： 2760250 查看本文章

0

作者: CHEN, Yue
单位: 浙江大学
时间限制: 150ms
内存限制: 64MB
代码长度限制: 16KB

代码：

num = input().split()
pa = ''
pb = ''
for i in num[0]:
    if num[1] == i:
        pa = pa + i
for j in num[2]:
    if num[3] == j:
        pb = pb + j
if pa == '':
    pa = 0
if pb == '':
    pb = 0
print(int(pa)+int(pb))