复旦18考研机试真题(2)--解方程

题目：给定一个字符串，代表一个一元一次方程。如果有解，输出格式“x=数字” ，如果解的个数无穷，输出 “infinite solutions”。 如果没有解输出“no solution”；字符串长度不超过 256。
样例:
输入:

10x-2x-8=4x+7+x
10x-2x-8=4x+x+7
-10x+20x-2x-8=4x+14+x-7
-10x+20x-2x-8=-4x+14+x-7+8x

输出:

x=5

分析：题目没说是否会出现乘除法、系数会不会是小数、没说结果是小数怎么办，没说是否会出现大数，没说是否会出现非法输入，有点坑；但是真正考试的时候，应该是可以和助教明确题目要求的，这里我们暂且不考虑大数，只考虑样例那样给出的合法输入，并且解只能是整数，否则认为无解。

这个题目考察的是字符串处理。按照’+’、’-‘以及’=‘来进行分隔；如果分割出的部分里有’x’，则为x的系数，否则是常数；当然有几种特殊情况要处理；第一，分割出的形如+x这样子的部分，系数要变化；第二，首尾两部分要特别考虑；

#include<iostream>
#include<string>   
using namespace std;
string a;   
int main() { 
	cin >> a;
	int len = a.length(), mark = 1,flag, b =0 , c =0;
	for (int i = 0; i < len; i++) { 
		if ((a[i] == '-'&&i!=0)|| a[i] == '+'|| a[i] == '='||i== len-1) {
			int j = i - 1;
			int tmp = (i == len - 1) ? i : i - 1;//考虑到尾部的情况
			while (j >= 0 && (a[j] != '-' && a[j] != '+'&&a[j]!='='))
				j--; 
			flag = (j >= 0 && a[j] == '-')? -1 : 1; //考虑到首部没有符号
			if (a[tmp] == 'x')
				b += tmp - j > 1 ? mark * flag*atoi(a.substr(j + 1, tmp - j).c_str()) : mark * flag;//考虑到形如+x这种情况
			else
				c += mark * flag*atoi(a.substr(j + 1, tmp - j).c_str());
		}
		if (a[i] == '=')//处理“=”右边的数据，逆号
			mark = -1;
	}  
	if (b == 0 && c == 0)
		printf("infinite solutions\n");
	else if (b == 0|| -c%b!=0)
		printf("no solution\n");
	else
		printf("x=%d\n", -c/ b);
	return 0;
}