PTA 1016 部分A+B(Python版)

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

输出格式：

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

输入样例 1：

3862767 6 13530293 3

输出样例 1：

399

输入样例 2：

3862767 1 13530293 8

输出样例 2：

0

提交结果： 

代码： 

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