Basic Level 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¹⁰。

输出格式：

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

输入样例1：

3862767 6 13530293 3

输出样例1：

399

输入样例2：

3862767 1 13530293 8

输出样例2：

0

C++

#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;

int GetNum(int base,int num){
    int r = 0,mul=1;
    for(int i=0;i<num;i++,mul*=10){
        r = r + base * mul;
    }
    return r;
}

int main() {
    char A[100]={0};
    char B[100]={0};
    int Da=0,Db=0;
    scanf("%s %d %s %d",A,&Da,B,&Db);
    int na = count(A,A+strlen(A),Da+'0');
    int nb = count(B,B+strlen(B),Db+'0');
    printf("%d\n",GetNum(Da,na)+GetNum(Db,nb));
    return 0;
}

Python

A, Da, B, Db = raw_input().split()
Pa, Pb = Da * list(A).count(Da), Db * list(B).count(Db)
Pa = int(Pa) if Pa!="" else 0
Pb = int(Pb) if Pb!="" else 0
print Pa + Pb