【PAT】1016 部分A+B（15 分）

1016 部分A+B（15 分）

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

输出格式：

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

输入样例 1：

3862767 6 13530293 3

输出样例 1：

399

输入样例 2：

3862767 1 13530293 8

输出样例 2：

0


C++代码如下：

#include<string>
#include<iostream>

using namespace std;
#define ll long long 

ll px( string a, int da) {
    ll sum=0;
    for (int i = 0; i < a.length(); i++) 
        if (a[i]-'0' == da) 
            sum = sum * 10 + da;
    return sum;
}
int main() {
    string a, b;
    int da, db;
    cin >> a >> da >> b >> db;
    ll sum_a, sum_b;
    sum_a = px(a, da);
    sum_b = px(b, db);
    cout << sum_a + sum_b << endl;
    return 0;
}