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的值。

输入样例1：

3862767 6 13530293 3

输出样例1：

399

输入样例2：

3862767 1 13530293 8

输出样例2：

0

//string转int可以使用atoi(string.c_str())

#include <iostream>
#include <string>
#include <algorithm>
#include <cstdlib>

using namespace std;

int countNum(string a, char da){
	int cnt = 0;
	for (int i = 0; i < a.length(); ++i){
		if (a[i] == da){
			cnt++;
		}
	}

	if (cnt){
		string  ans = "";
		for (int i = 0; i < cnt; ++i){
			ans += da;
		}
		return atoi(ans.c_str());
	}
	else{
		return cnt;
	}
}

int main(){

	string a, b;
	char da, db;
	cin >> a >> da >> b >> db;
	cout << countNum(a, da) + countNum(b, db)<<endl;
	system("pause");
	return 0;
}