PAT乙级练习题B1016.部分A+B

题目描述

正整数A的“DA（为1位整数）部分”定义为由A中所有DA组成的新整数PA。例如：给定A = 3862767，DA = 6，则A的“6部分”PA是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

代码

#include<iostream>
#include<vector>
#include<cmath>
using namespace std;

int take_part(int A, int Da)
{
    int i = 0;

    for (; A; A /= 10)
    {
        if (A % 10 == Da)
        {
            ++i;
        }
    }

    int sum = 0;
    for (; i > 0; --i)
    {
        sum += Da*pow(10,i-1);
    }

    return sum;

}


int main()
{
    int A, Da, B, Db;
    int PA , PB;

    cin >> A >> Da >> B >> Db;
    PA = take_part(A, Da);
    PB = take_part(B, Db);
    cout << PA + PB << endl;
    system("pause");
    return 0;

}