PAT B1016 部分A+B

题目地址：https://www.patest.cn/contests/pat-b-practise/1016

题目描述：

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

输出格式：

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

输入样例1：

3862767 6 13530293 3

输出样例1：

399

输入样例2：

3862767 1 13530293 8

输出样例2：

0

题意：

输入四个数 A、DA、B、DB，其中 DA 跟 DB 都是单个数字。将 A 中的数字 DA 全部拼在一起得到 PA，将 B 中的数字 DB 全都拼在一起得到 PB，输出 PA + PB。

样例解释：

样例 1：

A = 3862767，DA = 6 ;

B = 13530293 ,DB = 3

这样 6 在 A 中出现了两次，因此 PA = 66 ；而 3 在 B 中出现了三次，因此 PB = 333

最后输出 66 + 333 = 399。

样例 2：

A = 3862767，DA = 1；

B = 13530293，DB = 8.

这样 1 在 A 中出现了 0 次，因此 PA = 0；而 8 在 B 中出现了 0 次，因此 PB = 0。

最后输出 0 + 0 = 0。

解题思路：

令 PA 初值均为 0 ，枚举 A 中的每一位，如果该位恰好等于 DA，则令 PA = PA * 10 + DA，这样当枚举完 A 中的每一位之后，就得到了 PA。

同理，可以得到 PB。最后输出 PA + PB 即可。

注意：由于题目中给出的范围是10的10次方以内，这个范围是超过了 int 的，因此需要使用 long long 来存放 A 和 B。不过也可以用字符串来存储 A 和 B，方法其实都是一样的。

C++完整代码如下：

#include<cstdio>
int main(){
long long a, b, da, db;
scanf("%lld%lld%lld%lld", &a, &da, &b, &db);
long long pa = 0, pb = 0;
while(a != 0){      //枚举 a 的每一位
if(a % 10 == da) pa = pa * 10 + da; //若当前位为 da，给 pa 增加一位 da
a = a / 10;
}
while(b != 0){      //枚举 b 的每一位
if(b % 10 == db) pb = pb * 10 + db;
b = b / 10;
}
printf("%lld\n", pa + pb);
return 0;
}

python 完整代码如下：

#coding=utf-8
def count(a,b):
count=0
for i in a:
if i==b:
count=count*10+int(b)
return count
s=raw_input()
a,b,c,d=s.split(" ")
print count(a,b)+count(c,d)