【算法笔记】B1016 部分A+B

1016 部分A+B （15 分)

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

输出格式：

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

输入样例 1：

3862767 6 13530293 3

输出样例 1：

399

输入样例 2：

3862767 1 13530293 8

输出样例 2：

0

思路：

枚举P_A的每一位，如果与D_A相同，就执行 P_A = P_A * 10 +D_A。P_B同理可得，最后输出P_A+P_B。

为了简化代码定义一个函数getP()。注意题目的范围是10¹⁰，所以用long long定义A和B。

CODE：

#include<iostream>
using namespace std;

long long getP(long long a,int da){
    int i = 0, pa = 0;
    long long t = a;
    while(t){
        if(t % 10 == da){
            pa = pa * 10 + da; 
        }
        t /= 10;
    }
    return pa;
}

int main(){
    long long a, b;
    int da, db;
    cin>> a >> da >> b >> db;
    cout<< getP(a, da) + getP(b, db);
    return 0;
}