PAT乙级-1016 部分A+B (15 分)

1016 部分A+B (15 分)

正整数 AAA 的“DAD_AD​A​​（为 1 位整数）部分”定义为由 AAA 中所有 DAD_AD​A​​ 组成的新整数 PAP_AP​A​​。例如：给定 A=3862767A = 3862767A=3862767，DA=6D_A = 6D​A​​=6，则 AAA 的“6 部分”PAP_AP​A​​ 是 66，因为 AAA 中有 2 个 6。现给定 AAA、DAD_AD​A​​、BBB、DBD_BD​B​​，请编写程序计算 PA+PBP_A + P_BP​A​​+P​B​​。

输入格式：

输入在一行中依次给出 AAA、DAD_AD​A​​、BBB、DBD_BD​B​​，中间以空格分隔，其中 0<A,B<10100 < A, B < 10^{10}0<A,B<10​10​​。

输出格式：

在一行中输出 PA+PBP_A + P_BP​A​​+P​B​​ 的值。

输入样例 1：

3862767 6 13530293 3

输出样例 1：

399

输入样例 2：

3862767 1 13530293 8

输出样例 2：

0

代码

#include<iostream>
using namespace std;
int main(){
 int A,Da,B,Db;
 int result1=0,result2=0;
 cin>>A>>Da>>B>>Db;
 int temp=A,step=1;
 while(temp != 0){
  temp = A%10;
  if(temp == Da){
   result1 = result1+(temp*step);
   step*=10;
  }
  A = A/10;
 }
 temp=B;step=1;
 while(B != 0){
  temp = B%10;
  if(temp == Db){
   result2 = result2+(temp*step);
   step*=10;
  }
  B = B/10;
 }
 cout<<result1+ result2<<endl;
 return 0;
}

运行结果：

答案正确。