算法入门模拟-部分A+B

题源：题源：PAT B1016
正整数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 < 10¹⁰。
输出格式：
在一行中输出PA + PB的值。

输入样例：
3862767 6 13530293 3
输出样例：
399

输入样例2：
3862767 1 13530293 8
输出样例2：
0

思路：问题关键点在于如何根据DA求得PA。用%方法取A的个位十位百位…与DA比较，相等则在原来的数值上*10+DA。

#include <cstdio>
long long fun(long long A,int DA){
	long long  PA=0;
	while(A!=0){
		if(A%10==DA) PA=PA*10+DA;
		A=A/10;
	}

	return PA;
}
int main(){
	long long A,B,PA,PB;
	int DA,DB;
	scanf("%lld%d%lld%d",&A,&DA,&B,&DB);
	PA=fun(A,DA);
	PB=fun(B,DB);
	printf("%lld",PA+PB);
	return 0;
}