PAT乙 1016. 部分A+B (15)

题目地址：PAT乙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的值。

程序：

#include <iostream>
#include <string>

using namespace std;

int getAnswer(int DA, long A)
{
	int reg = 0;
	while (A)
	{
		int temp = A%10;
		A /= 10;
		if (DA == temp)
		{
			reg *= 10;
			reg += DA;
		}
	}
	return reg;
}


int main()
{
	long A, B;
	int DA, DB;
	cin >> A >> DA >> B >> DB;

	cout << getAnswer(DA, A) + getAnswer(DB, B) << endl;
}