PAT乙级练习题_1016“部分A+B”_python解题

原题

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

my answer

思路

在第一个输入样例中，11 * 6 = 66
所以创建一个变量pre_a = "0"
遍历整个字符串，在遇到相同的数字时，pre_a += "1"，遍历完之后，将pre_a转成int类型并乘上响应的数字，即可。然后将两个数相加。

input_str = input()
ls = input_str.split()
A = ls[0]
Da = ls[1]
B = ls[2]
Db = ls[3]

pre_A = "0"
for i in A:
    if i == Da:
        pre_A += "1"

pre_B = "0"
for i in B:
    if i == Db:
        pre_B += "1"

sum_A = int(pre_A) * int(Da)
sum_B = int(pre_B) * int(Db)
print(sum_A + sum_B)