P1022 计算器的改良

题目背景

NCL是一家专门从事计算器改良与升级的实验室，最近该实验室收到了某公司所委托的一个任务：需要在该公司某型号的计算器上加上解一元一次方程的功能。实验室将这个任务交给了一个刚进入的新手ZL先生。

题目描述

为了很好的完成这个任务，ZLZL先生首先研究了一些一元一次方程的实例：

4+3x=84+3x=8

6a-5+1=2-2a6a−5+1=2−2a

-5+12y=0−5+12y=0

ZLZL先生被主管告之，在计算器上键入的一个一元一次方程中，只包含整数、小写字母及+、-、=这三个数学符号（当然，符号“-”既可作减号，也可作负号）。方程中并没有括号，也没有除号，方程中的字母表示未知数。

你可假设对键入的方程的正确性的判断是由另一个程序员在做，或者说可认为键入的一元一次方程均为合法的，且有唯一实数解。

输入格式

一个一元一次方程。

输出格式

解方程的结果(精确至小数点后三位)。

输入输出样例

6a-5+1=2-2a
a=0.750

题目易错：

1、我是在不是数字的时候才对数字进行存储，所以在最后时需要特别判断一下以数字结尾的情况
2、需要考虑未知数系数为1时

AC代码

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
#include<set>
#include<sstream>
#include<map>
#include<stack>
#include<queue>
#define LL long long
using namespace std;

int a[100005];
int x[100005];
int main() {
    int num = 0;
    int cnt1 = 0;
    int cnt2 = 0;
    string s;
    cin>>s;
    int flag = 1;
    int j = 1;
    int n = s.length();
    char c;
    for(int i = 0;i < n;i++) {
        if(s[i] >= '0' && s[i] <= '9') {
            num = num * 10 + s[i] - '0';
        } else {
            if(s[i] >= 'a' && s[i] <= 'z') {
                if(num == 0)
                    num = 1;
                c = s[i];
                x[cnt1++] = num * flag * j;
                num = 0;
            }  else {

                    a[cnt2++] = num * flag * j;
                    j = 1;
                    num = 0;
                    if(s[i] == '-')
                        j = -1;
                    if(s[i] == '=') {
                        flag = -1;
                    }
                }
        }
        if(i == n - 1&&s[i] >= '0' && s[i] <= '9')
            a[cnt2++] = num * flag * j;
    }

    int x1 = 0;
    int x2 = 0;
    for(int i = 0;i < cnt1;i++)
        x1 += x[i];
        //cout<<x[i]<<endl;
    for(int i = 0;i < cnt2;i++)
        x2 += a[i];
        //cout<<a[i]<<endl;
    //cout<<x1<< " "<<x2<<endl;
    double x3 = (double)x1 * -1;
    double x4 = (double)x2;
    double ave = x4 / x3;
    printf("%c=%.3lf\n", c, ave);
    return 0;
}