【PAT乙级】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

解题思路：

保存$A$为字符串，枚举字符串的每一位，匹配关键字$D_A$，匹配成功则用公式 $P_A=P_A\ast 10+D_A$计算，字符串$B$同理。

答案代码：

#include <iostream>
#include <string>
using namespace std;

int search(string s, int key) {
    int num = 0;
    for (int i = 0; i != s.size(); i++) {   // 枚举字符串每一位
        if (s[i] - '0' == key)              // 匹配关键字
            num = num*10 + key;            
    }
    return num;
}

int main() {
    string a, b;
    int da = 0, db = 0, m =0, n = 0;
    cin >> a >> da >> b >> db;
 
    m = search(a, da);
    n = search(b, db);
    
    cout << m + n << endl;
    
    return 0;
}