PAT乙 1016 部分A+B C++

PAT乙 1016 部分A+B

题目
正整数 A 的DA（为 1 位整数）部分定义为由 A 中所有 DA​​ 组成的新整数 P​A。例如：给定 A=3862767，DA​​ =6，则 A 的“6 部分”P​A是 66，因为 A 中有 2 个 6。
现给定 A、D​A​​ 、B、DB，请编写程序计算 P​A​​ +P​B​​ 。
输入格式：
输入在一行中依次给出 A、D​A​ 、B、DB ，中间以空格分隔，其中 0<A,B<10​^10。
输出格式：
在一行中输出 PA​​ +PB 的值。
​​
思路
将正整数A作为字符串存入，然后遍历寻找其中DA的个数。计算出出PA的值，同理计算PB的值，相加输出。
代码

#include <string>
#include <iostream>
using namespace std;
int fun(char DA,int Numa){//根据DA和和DA的个数的值求PA
	int i=DA-'0';
	int PA=0;
	while(Numa!=0){
	   PA=PA*10+i;
	   Numa--;	
	}
	return PA;
}
int main(){
	string A,B;
	char DA,DB;
	int Numa,Numb;
	Numa=Numb=0;
	cin>>A>>DA>>B>>DB;
	for(int i=0;i<A.length();i++){//寻找DA的个数
		if(A[i]==DA)
		Numa++;
	}
	for(int i=0;i<B.length();i++){//寻找DB的个数
		if(B[i]==DB)
		Numb++;
	}
	cout<<(fun(DA,Numa)+fun(DB,Numb));
return 0;
}