复旦18年机考-求一元一次方程的解（string）

题意是输入一个一元一次方程的字符串，求解其解，解可能是有解，无穷多解（恒等式），无解（x的系数为零，常数项不为0）
解题思路：分别统计x前面的系数和常数项即可，但是这也很麻烦，具体代码见下面。

#include <iostream>
#include <string>
#include <cstring>

using namespace std;

//10x-2x-8=4x+7+x

int main()
{
	string str;
	int num[100];
	int mark[100];
	while(getline(cin, str))
	{
		memset(num, 0, sizeof(num));
		memset(mark, 0, sizeof(mark));
		int j = 0;
		int tmp;  //标记num数组中，从哪个数开始是等式右边的 
		for(int i = 0; i < str.length(); i++)
		{
			while(str[i] >= '0' && str[i] <= '9')
			{
				num[j] = num[j] * 10 + str[i] - '0';
				i++;
			}
			//处理x+2x=6这种第一个x的系数就是1的情况 
			if(i == 0 && str[i] == 'x')
				num[j] = 1;	
			//处理其他位置x系数为1的情况 
			if(str[i] == '-' || str[i] == '+' || str[i] == '=' || str[i] == 'x')
				if(str[i+1] == 'x')
					num[j+1] = 1;
				
			if(mark[j] == 1)
				num[j] = -num[j];
			if(str[i] == '-')     //num[j+1]是负数 
				mark[j+1] = 1;
			if(str[i] == 'x')     //表示是x前面的系数 
				mark[j] = 2;     
			if(str[i] == '=')    //标记“=”右边第一个数字的位置 
				tmp = j+1;
			j++;
		}
		int x1 = 0, x2 = 0;   //分别表示系数和常数 
		for(int k = 0; k < j; k++)
		{
			if(mark[k] == 2 && k < tmp)
				x1 += num[k];
			if(mark[k] == 2 && k >= tmp)
				x1 -= num[k];
			if(mark[k] < 2 && k < tmp)
				x2 -= num[k];
			if(mark[k] < 2 && k >= tmp)
				x2 += num[k];
		}
		if(x1 == 0 && x2 == 0)
			printf("infinite solutions\n");
		else if(x1 == 0 && x2 != 0)
			printf("no solutions\n");
		else	
			printf("%d %d %d\n", x1, x2, x2/x1);
	} 
	return 0;
}