弟弟的作业——dengdengoj

题目描述
你的弟弟刚做完了“100以内数的加减法”这部分的作业，请你帮他检查一下。每道题目（包括弟弟的答案）的格式为a+b=c或者a-b=c，其中a和b是作业中给出的，均为不超过100的非负整数；c是弟弟算出的答案，可能是不超过200的非负整数，也可能是单个字符"?"，表示他不会算。

输入
输入文件包含不超过100行，以文件结束符结尾。每行包含一道题目，格式保证符合上述规定，且不包含任何空白字符。输入的所有整数均不含前导0。

输出
输出仅一行，包含一个非负整数，即弟弟答对的题目数量。

样例输入
1+2=3
3-1=5
6+7=?
99-0=99
样例输出
2

AC代码：

#include <iostream>
#include <stdio.h>
#include <string>
using namespace std;

int main() {
	string str;
	int cnt = 0; //答对的数量
	bool flag;
	while (cin >> str) {
		int len = str.length();
		if (str[len - 1] == '?') {
			continue;
		}
		else {
			int k = 0;
			if (str.find('+') != string::npos) {
				flag = true;
				k = str.find('+');
			}
			else {
				flag = false;
				k = str.find('-');
			}
			int a = 0, b = 0, c = 0;
			for (int i = 0; i < k; i++) {
				a = a * 10 + str[i] - '0';
			}
			for (int i = k + 1; i < str.find('='); i++) {
				b = b * 10 + str[i] - '0';
			}
			for (int i = str.find('=') + 1; i < str.length(); i++) {
				c = c * 10 + str[i] - '0';
			}
			if (flag) {
				if (a + b == c) {
					cnt++;
				}
			}
			else {
				if (a - b == c) {
					cnt++;
				}
			}
		}
	}
	cout << cnt << endl;
	return 0;
}