[PAT乙级] Practise 1016 部分A+B

PAT (Basic Level) Practice （中文）1016

1016 部分A+B

正整数 A 的“*D**A（为 1 位整数）部分”定义为由 A* 中所有 *D**A* 组成的新整数 PA。例如：给定 A=3862767，DA=6，则 A 的“6 部分”*P**A* 是 66，因为 A 中有 2 个 6。

现给定 A、DA、B、DB，请编写程序计算 PA+PB。

输入格式：

输入在一行中依次给出 A、DA、B、DB，中间以空格分隔，其中 0<A,B<1010。

输出格式：

在一行中输出 PA+PB 的值。

输入样例 1：

3862767 6 13530293 3

输出样例 1：

399

输入样例 2：

3862767 1 13530293 8

输出样例 2：

0

思路：

根据题意计算串中的重复数字并根据Da Db构造新的字符串即可。

代码：

#include <iostream>
#include <string>
using namespace std;
int inWath(string s,char k) {
    int n = 0;
    for (int i = 0; i < s.size(); i++) {
        if (s[i] == k)
            n++;
    }
    return n;
}
int main() {
    string a, b;
    char   Da,Db;
    string Pa,Pb;
    cin >> a >> Da >> b >> Db;
    if (inWath(a, Da) != 0 && inWath(b, Db) != 0)
        cout << stoi(Pa.insert(0, inWath(a, Da), Da)) + stoi(Pb.insert(0, inWath(b, Db), Db));
    else
        cout << "0";
    return 0;
}