PAT BASIC LEVEL 1016. 部分A+B (15)

1016. 部分A+B (15)

正整数A的“D_A（为1位整数）部分”定义为由A中所有DA组成的新整数PA。例如：给定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 < 1010。

输出格式：

在一行中输出P_A + P_B的值。
输入样例1：

3862767 6 13530293 3

输出样例1：

399

输入样例2：

3862767 1 13530293 8

输出样例2：

0

#include<iostream>
using namespace std;

int get_sum(char* numbers, int key) {
    int sum = 0;
    while(*numbers) {
        if(*numbers - '0' == key)
            sum = sum * 10 + key;
        numbers++;
    }
    return sum;
}

int main() {
    int sum[2];
    for(int i = 0;i < 2;i ++) {
        char numbers[11];
        int key;
        cin >> numbers >> key;
        sum[i]   = get_sum(numbers, key);
    }
    cout << sum[0] + sum[1];
}

PS.
只是习惯性的写个PS，没什么要说的。