《算法笔记》3.1小节——入门模拟-＞简单模拟 问题 H: 部分A+B (15)

问题 H: 部分A+B (15)

时间限制 : 1.000 sec 内存限制 : 32 MB

题目描述

正整数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 < 10¹⁰。

输出

在一行中输出P_A+ P_B的值。

样例输入

3862767 6 13530293 3
3862767 1 13530293 8

样例输出

399
0

此题和PAT乙级1016相同

#include <bits/stdc++.h>

using namespace std;
using gg = long long;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    string a, b;
    gg a_num, b_num, sumA = 0, sumB = 0;
    cin >> a >> a_num >> b >> b_num;
    for (auto c:a)
        if (c - '0' == a_num)
            sumA = sumA * 10 + a_num;
    for (auto c:b)
        if (c - '0' == b_num)
            sumB = sumB * 10 + b_num;
    cout << sumA + sumB;
    return 0;
}