【PAT天梯赛】团体程序设计天梯赛 L1-025 正整数A+B (15分) 这题输入有点坑

题目

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

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

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

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

输入样例1：
123 456
输出样例1：
123 + 456 = 579
输入样例2：
22. 18
输出样例2：
? + 18 = ?
输入样例3：
-100 blabla bla…33
输出样例3：
? + ? = ?

思路

这题对于b的输入非常坑，A与B之间会有空格隔开，也就是说第一个空格之前都属于A，空格之后都属于B，但是B本身是可以带空格的，如果直接用cin就WA了，然后因为A和B的输入其实是不清楚的，如果字符串里没有数字的话直接用stoi（）这个函数就会报错，所以对于什么时候用stoi（）的位置要考虑一下，我直接把最后要输出的内容放在一个变量中，这样整体思路就比较清晰了。

代码

#include<bits/stdc++.h>
using namespace std;
int main() {
	string a, b, a2, b2, a3, b3;
	char c;
	int ans;
	cin >> a;
	getchar();
	while((c = cin.get()) != '\n') b = b + c;
	for(int i = 0; i < a.size(); i++) {
		if(a[i] >= '0' && a[i] <= '9') {
			a2 = a2 + a[i];
		}
	}
	for(int i = 0; i < b.size(); i++) {
		if(b[i] >= '0' && b[i] <= '9') {
			b2 = b2 + b[i];
		}
	}
	if(a2 == a && stoi(a2) <= 1000 && stoi(a2) >= 1) a3 = a2;
	else a3 = "?";
	if(b2 == b && stoi(b2) <= 1000 && stoi(b2) >= 1) b3 = b2;
	else b3 = "?";
	cout << a3 << " + " << b3 << " = ";
	if(a3 == "?" || b3 == "?") cout << "?";
	else cout << stoi(a2) + stoi(b2);
	return 0;
}