【PAT 乙级】1016 部分A+B

1016 部分A+B (15)（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^10^。

输出格式：

在一行中输出P~A~ + P~B~的值。

输入样例1：

3862767 6 13530293 3

输出样例1：

399

输入样例2：

3862767 1 13530293 8

输出样例2：

0

题目也没什么坑，应该有内存和耗时更小的写法，

反正我的代码如下：

#include <iostream>
#include <string>
using namespace std;
int main()
{
	char da,db;
	string a,b;
	int lenA,lenB,len;
	long long Pa=0,Pb=0,sum=0;

	cin >>a>>da>>b>>db;

	lenA = a.size();
	lenB = b.size();
	len = (lenA>lenB)?lenA:lenB;
	for(int i=0;i<len;i++)
	{
		if(a[i]==da && i<lenA)
		{
			Pa=Pa*10+(da-'0');
		}
		if(b[i]==db && i<lenB)
		{
			Pb=Pb*10+(db-'0');
		}
	}

	sum = Pa+Pb;
	cout <<sum;

	system("pause");
	return 0;
}