PAT B级 1016部分A+B (15point(s))

题目

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

现给定 A、DA、B、D​B ，请编写程序计算 P​A +P​B 。

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

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

输入样例 1：

3862767 6 13530293 3

输出样例 1：

399

输入样例 2：

3862767 1 13530293 8

输出样例 2：

0

思路

刚开始做的时候，用 long 接收的 A, B，所以自然想到用取余的形式取出每一位，但是看了柳婼的解法，恍然大悟，char 转 int 可以用 char - ‘0’ 来做，string 转 int 就可以用 stoi(string)

代码

用 long 接收的做法：

#include <iostream>

using namespace std;

int main() {
    long A, B;
    int da, db;

    cin >> A >> da >> B >> db;

    int arr1[10] = {0};

    while (A != 0 || A / 10 != 0) {
        arr1[A % 10]++;
        A = A / 10;
    }

    int arr2[10] = {0};

    while (B != 0 || B / 10 != 0) {
        arr2[B % 10]++;
        B = B / 10;
    }

    int sum1 = 0, sum2 = 0;
    for (int i = 0; i < arr1[da]; i++) {
        sum1 = sum1 * 10 + da;
    }

    for (int i = 0; i < arr2[db]; i++) {
        sum2 = sum2 * 10 + db;
    }

    cout << sum1 + sum2;
}

用 string 的做法：

#include <iostream>

using namespace std;

int main() {
    string A, B;
    int da, db;
    int pa = 0, pb = 0;
    cin >> A >> da >> B >> db;
    for (char c : A) {
        if (c - '0' == da) {
            pa = pa * 10 + da;
        }
    }

    for (char c : B) {
        if (c - '0' == db) {
            pb = pb * 10 + db;
        }
    }
    cout << pa + pb;
}