PAT乙级1016.部分A+B（15 分）

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

输出格式：

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

输入样例 1：

3862767 6 13530293 3

输出样例 1：

399

输入样例 2：

3862767 1 13530293 8

输出样例 2：

0

#include<iostream>
#include<cmath>
#include<string>
#include<stdlib.h>
using namespace std;

int count(string s, char a)
{
    int num = 0;
    for (int i = 0; i < s.length(); i++)
    {
        if (s[i] == a)
            num++;
    }
    return num;
}

long P(char a, int num)
{
    long sum = 0;
    while (num--)
    {
        sum += pow(10, num) * (a - '0');
    }
    return sum;
}
int main()
{
    string A, B;
    char DA, DB;
    int numA, numB;
    long PA, PB;
    cin >> A >> DA >> B >> DB;
    numA = count(A, DA);
    numB = count(B, DB);
    PA = P(DA, numA);
    PB = P(DB, numB);
    cout << PA + PB << endl;
    system("pause");
    return 0;
}