PAT乙级 1016 部分A+B

一.题目描述

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

输出格式：

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

二.代码

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int A,Da,B,Db;
    cin>>A>>Da>>B>>Db;
    int i=0,r=0;
    while(A)
    {
        int m=A%10;
        if(m==Da)
        {
            r+=m*pow(10,i);
            i++;
        }
        A/=10;
    }
    i=0;
    while(B)
    {
        int m=B%10;
        if(m==Db)
        {
            r+=m*pow(10,i);
            i++;
        }
        B/=10;
    }
    cout<<r;
}