C++/Python PAT乙级 1016 部分A+B (15)

题目描述

正整数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的值。

输入例子:

3862767 6 13530293 3

输出例子:

399

C++

#include <iostream>
#include <cmath>
#include <iomanip>
//#include <cstring>
//#include <cstdio>
//#include <algorithm>
using namespace std;

int Tiqu(unsigned int num, int result[])
{
	int count = 0;
	while(1)
	{
		result[count ++ ] = num % 10;
		num /= 10;
		if(num == 0) break;
	}
	return count;
}
int hebing(int result[],int count1,int D)
{
    int num=0,P=0;
    for (int i=0;i<count1;i++)
    {
        if (D==result[i])
            {
                P=P+D*pow(10,num);
                num++;
             }
    }
    return P;
}

int main()
 {
     const int N=pow(10,5);
     int A,B,DA,DB;
     cin >> A >> DA >> B >> DB;
     int result_A[N],result_B[N];
     int count_A=Tiqu(A,result_A);
     int count_B=Tiqu(B,result_B);
     int PA=hebing(result_A,count_A,DA);
     int PB=hebing(result_B,count_B,DB);
     cout << PA+PB ;
     return 0;
 }

Python

a,da,b,db=input().split()
def prinum(x,dx):
    num=0
    count=0
    for i in range(len(x)):
        if x[i]==dx:
            count+=1
    if count>0:
        num=int(dx*count)
    return num
a=list(a)
b=list(b)
print(prinum(a,da)+prinum(b,db))