PTA 正整数A+B(15分)

正整数A+B

题的目标很简单，就是求两个正整数A和B的和，其中A和B都在区间[1,1000]。稍微有点麻烦的是，输入并不保证是两个正整数。

输入格式：

输入在一行给出A和B，其间以空格分开。问题是A和B不一定是满足要求的正整数，有时候可能是超出范围的数字、负数、带小数点的实数、甚至是一堆乱码。

注意：我们把输入中出现的第1个空格认为是A和B的分隔。题目保证至少存在一个空格，并且B不是一个空字符串。

输出格式：

如果输入的确是两个正整数，则按格式A + B = 和输出。如果某个输入不合要求，则在相应位置输出?，显然此时和也是?。

输入样例1：

123 456

Copy

输出样例1：

123 + 456 = 579

Copy

输入样例2：

22. 18

Copy

输出样例2：

? + 18 = ?

Copy

输入样例3：

-100 blabla bla...33

Copy

输出样例3：

? + ? = ?

#include<bits//stdc++.h>
using namespace std;
string a,b;int flag=1,sum=0;
void solve(string &a)
{
    int num=atoi(a.c_str());
    if(a.size()<=4&&num>=1&&num<=1000&&to_string(num)==a)
    {
        sum+=num;
    }
    else 
    {
        a="?";
        flag=0;
    }
}
int main()
{
    getline(cin,a);
    string b=a;
    if(a[0]==' ')
    {
        a='?';
        b.erase(0,b.find_first_of(" ")+1);
        solve(b);
        cout<<a<<" + "<<b<<" = ";
        if(flag) cout<<sum;
        else cout<<"?";
    }
    else 
    {
    b.erase(0,b.find_first_of(" ")+1);
    a.erase(a.find_first_of(" "));
    solve(a);solve(b);
    cout<<a<<" + "<<b<<" = ";
    if(flag) cout<<sum;
    else cout<<"?";
    }
    return 0;
}