原题
正整数 A 的“ D A D_A DA(为 1 位整数)部分”定义为由 A 中所有 D A D_A DA 组成的新整数 P A P_A PA。例如:给定 A=3862767, D A D_A DA=6,则 A 的“6 部分” P A P_A PA 是 66,因为 A 中有 2 个 6。
现给定 A、 D A D_A DA、B、 D B D_B DB,请编写程序计算 P A P_A PA+ P B P_B PB。
输入格式
输入在一行中依次给出 A、 D A D_A DA、B、 D B D_B DB,中间以空格分隔,其中 0<A,B< 1 0 10 10^{10} 1010。
输出格式
在一行中输出 P A P_A PA+ P B P_B PB 的值。
输入样例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)