PAT B1016 部分A+B (15point(s))

正整数 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

• 思路：
  首先，A，B范围为10¹⁰， 超过int 范围，所以有两个选择，一是string（code1），二是long long（code2）
  step1：得到P_A ：遍历A的每一位，每遇到D_A就累加一位D_A(可通过字符str += c;，也可不断*10迭代p = p * 10 + da)

• code1:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll GetP(string s, int d){
	ll p = 0;
	for(int i = 0; i < s.size(); ++i){
		if(s[i]-'0' == d){
			p = p * 10 + d;	
		} 
	}
//	for(auto c : s){
//		if(c-'0' == d){
//			p = p * 10 + d;
//		}
//	}
	return p;
}

int main(){
	string a, b;
	int da, db;
	cin >> a >> da >> b >> db;
	cout << GetP(a, da) + GetP(b, db);
	return 0;
}

• code2：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll GetP(ll a, int d){
	ll p = 0;
	while(a){
		if(a % 10 == d) p = p * 10 + d;
		a /= 10;
	}
	return p;
}

int main(){
	ll a, b;
	int da, db;
	cin >> a >> da >> b >> db;
	cout << GetP(a, da) + GetP(b, db);
	return 0;
}