部分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。

   

#include<iostream>
#include<string>
using namespace std;


int sumAandB(string strA, int DA, string strB, int DB){
	int sumA = 0;
	int sumB = 0;
	for (int i = 0; i < strA.size(); i++){
		char tmp = strA[i];
		if ((tmp - '0') == DA)
			sumA = sumA * 10 + (tmp - '0');
	}
	for (int i = 0; i < strB.size(); i++){
		char tmp = strB[i];
		if ((tmp - '0') == DB)
			sumB = sumB * 10 + (tmp - '0');
	}
	return sumA + sumB;
}

int main(){
	string strA, strB;
	int DA, DB;
	cin >> strA >> DA >> strB >> DB;
	cout << sumAandB(strA, DA, strB, DB) << endl;
	cin.get();
	cin.get();
}