【PAT乙級】部分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​10​​。

輸出格式：

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

輸入樣例 1：

3862767 6 13530293 3

輸出樣例 1：

399

輸入樣例 2：

3862767 1 13530293 8

輸出樣例 2：

0

代碼：

#include<iostream>
using namespace std;
long fun(long A,int Da);
int main() {
    long A,B,X,Y;
    int Da,Db;
    cin>>A>>Da>>B>>Db;
    X=fun(A,Da);
    Y=fun(B,Db);
    cout<<X+Y;
    return 0;
}
long fun(long A,int Da) {
    int S;long sum=0;
    while(A) {
        S=A%10;
        A=A/10;
        if(S==Da)
            sum=10*sum+S;
    }
    return sum;
}