C程序-PTA-1016 部分A+B

正整数 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 <iostream>
#include <stdio.h>


int solve(long long N,int n);

int main( ) 
{
	long long A,B;
	int a,b;
	scanf("%lld%d%lld%d",&A,&a,&B,&b);
	
	printf("%d",solve(A,a)+solve(B,b));
	
	return 0;
}
int solve(long long N,int n)
{
	int p=0;
	while(N>0)
	{
		if(N%10==n)
		{
			p=p*10+n;
		}
		N=N/10;
	}
	return p;
}