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<1010。

输出格式：

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

输入样例 1：

3862767 6 13530293 3

      
    

输出样例 1：

399

      
    

输入样例 2：

3862767 1 13530293 8

      
    

输出样例 2：

0

代码

// 1016 部分A+B.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//

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

long getp(string a, int da) {
    int cnt = 0;
    for (int i = 0; i < a.length(); i++) {
        if (a[i] - '0' == da) {
            cnt++;
        }
    }
    if (cnt == 0)
        return 0;
    string str;
    for (int i = 0; i < cnt; i++) {
        str += to_string(da);
    }
    return stol(str);
}

int main()
{
    string a, b;
    int da, db;
    cin >> a >> da >> b >> db;
    cout << getp(a, da) + getp(b, db);
    return 0;
}