【pat乙級】部分A+B

(pat乙級1016)

正整數 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<iostream> 
using namespace std;
int main(){
    string a,b;
    int da,db;
    cin>>a>>da>>b>>db; 
    int la=a.length(),lb=b.length();
    long long sum=0,aa=1,bb=1;
    for(int i=la;i>=0;i--){
        if((a[i]-'0')==da){
            sum+=aa*(a[i]-'0');
            aa*=10;
        }
    }
    for(int i=lb;i>=0;i--){
        if(b[i]-'0'==db){
            sum+=bb*(b[i]-'0');
            bb*=10;
        }
    }
    cout<<sum;
    return 0; 
}

注意：
10 ^10>2 ^32，輸入的a，b聲明爲string類型方便遍歷每一位，之和sum，乘數aa，bb聲明爲long long類型防止超範圍。