PAT 1016 部分A+B C语言

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

#include <stdio.h>
#include <stdlib.h>
#include <math.h>

int number(int a, int b);
int newNum(int num, int repeatTime);
int main()
{
    int a,b, la, lb;
    scanf("%d%d%d%d", &la,&a,&lb,&b);

    int a1 = number(la,a);
    int b1 = number(lb, b);

    int new_a = newNum(a, a1);
    int new_b = newNum(b,b1);
    printf("%d",new_a +new_b);

    return 0;
}

int number(int a, int b){
    int cnt = 0;
    while( a != 0){
        if(a % 10 == b) cnt++;
        a /= 10;
    }
    if ( a == b) cnt ++;
     return cnt;
}

int newNum(int num, int repeatTime){
    int result = 0,i = 0;
    for(; i < repeatTime; i++){
        result += num * (pow(10,i));
    }

    return result;
}