PAT 乙级 1016 部分A+B

正整数 A 的“DA​（为 1 位整数）部分”定义为由 A 中所有 DA​ 组成的新整数 PA​。例如：给定 A=3862767，DA​=6，则 A 的“6 部分”PA​ 是 66，因为 A 中有 2 个 6。

现给定 A、DA​、B、DB​，请编写程序计算 PA​+PB​。

输入格式：

输入在一行中依次给出 A、DA​、B、DB​，中间以空格分隔，其中 0<A,B<109。

输出格式：

在一行中输出 PA​+PB​ 的值。

 Python

num = input().split()
pa = ''
pb = ''

for i in range(len(num[0])):
    if num[0][i] == num[1]:
        pa = pa + num[1]
for i in range(len(num[2])):
    if num[2][i] == num[3]:
        pb = pb + num[3]

if pa and pb:
    print(int(pa) + int(pb))
elif not pa and not pb:
    print(0)
else:
    print(int(pa+pb))