1016. 部分A+B (15)

时间限制 1000 ms 内存限制 32768 KB 代码长度限制 100 KB 判断程序 Standard (来自 小小)
题目描述
正整数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的值。
输入例子:
3862767 6 13530293 3
输出例子:
399

#include <cstdio>
#include <iostream>
#include <cstring>
#include <cstdlib>
using namespace std;

long long my_atoll(string s)
{
    long long result = 0;
    char *p = &s[0];

    while(*p != '\0')
    {
        if( (*p >= '0') && (*p <= '9') )
        {
            result = result * 10 + *p - '0';
            p++;
        }
    }
    return result;
}

int main(int argc, char **argv)
{
    string strA, strB;
    int da, db;
    cin >> strA >> da >> strB >> db;
    string strC;
    string strD;
    for(int i=0; strA[i] != '\0'; i++)
    {
        if(strA[i] - '0' == da)
        {
            strC += strA[i];
        }
    }
    for(int i=0; strB[i] != '\0'; i++)
    {
        if(strB[i] - '0' == db)
        {
            strD += strB[i];
        }
    }

    cout << my_atoll(strC) + my_atoll(strD) << endl;
}