ZJU PAT 1016 部分A+B

1016. 部分A+B (15)

时间限制

100 ms

内存限制

32000 kB

代码长度限制

8000 B

判题程序

Standard

作者

CHEN, Yue

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

long long getRes(long long a, int da)
{
	long long tmp = 0;
	while(a)
	{
		if(a % 10 == da)
			tmp = tmp * 10 + da;
		a /= 10;
	}
	return tmp;
}

int main()
{
	long long a, b;
	int da, db;
	scanf("%lld %d %lld %d", &a, &da, &b, &db);
	printf("%lld\n", getRes(a, da) + getRes(b, db));
	return 0;
}