PAT 1016 部分A+B

https://pintia.cn/problem-sets/994805260223102976/problems/994805306310115328

正整数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

 

代码：

#include <bits/stdc++.h>

using namespace std;

const int maxn=1e5+10;
char a[maxn],b[maxn];

int main()
{
    int A,B;
    int cnt1=0,cnt2=0,sum1=0,sum2=0;
    scanf("%s%d%s%d",a,&A,b,&B);
    int len1=strlen(a);
    int len2=strlen(b);
    for(int i=0;i<len1;i++)
    {
        if(a[i]-'0'==A) cnt1++;
    }
    for(int i=0;i<len2;i++)
    {
        if(b[i]-'0'==B) cnt2++;
    }
    for(int i=1;i<=cnt1;i++)
    {
        sum1=sum1*10+A;
    }
    for(int i=1;i<=cnt2;i++)
    {
        sum2=sum2*10+B;
    }
    int sum=0;
    sum=sum1+sum2;
    //cout<<sum1<<endl<<sum2<<endl;
    printf("%d",sum);
    return 0;
}