B1016 部分A+B

1016 部分A+B （15 分)

正整数 A 的“DA（为 1 位整数）部分”定义为由 A 中所有 DA 组成的新整数 PA。例如：给定 A=3862767，DA=6，则 A 的“6 部分”PA 是 66，因为 A 中有 2 个 6。

现给定 A、DA、B、DB，请编写程序计算 PA+PB。

输入格式：

输入在一行中依次给出 A、DA、B、DB，中间以空格分隔，其中 0<A,B<1010。

输出格式：

在一行中输出 PA+PB 的值。

输入样例 1：

3862767 6 13530293 3

输出样例 1：

399

输入样例 2：

3862767 1 13530293 8

输出样例 2：

0

参考代码

#include <iostream>
#include <string>
#include <algorithm>
using namespace std;
int main(){
  string str1, str2;
  char a, b;
  int sum1=0, sum2=0;
  cin>>str1>>a>>str2>>b;
  int n=count(str1.begin(), str1.end(), a);
  int m=count(str2.begin(), str2.end(), b);
  if(n==0 && m==0) cout<<0;
  else{
    int t1=a-'0';
    int t2=b-'0';
    for(int i=0;i<n;i++){
      sum1=t1+sum1*10;
    }
    for(int i=0;i<m;i++){
      sum2=t2+sum2*10;
    }
    cout<<sum1+sum2;
  }
  return 0;
}