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

输出格式：

在一行中输出 PA​+PB​ 的值。

输入样例 1：

3862767 6 13530293 3

输出样例 1：

399

输入样例 2：

3862767 1 13530293 8

输出样例 2：

0

 我的答案：（仅供参考）

#include
int part(int a,int b);
int main(){
    int sum,a1,a2,b1,b2;
    scanf("%d %d %d %d",&a1,&a2,&b1,&b2);
    sum=part(a1,a2)+part(b1,b2);
    printf("%d",sum);
    return 0;
}
int part(int a,int b){
    int n,count=0;
    while(a>0){
       n=a%10;
       if(n==b){count++;}
       a/=10;
    }
    switch(count){
        case 0:n=0;break;
        case 1:n=b;break;
        case 2:n=b*11;break;
        case 3:n=b*111;break;
        case 4:n=b*1111;break;
        case 5:n=b*11111;break;
        case 6:n=b*111111;break;
        case 7:n=b*1111111;break;
        case 8:n=b*11111111;break;
        case 9:n=b*111111111;break;
    }return n;
}