1018: 计算器的改良（简单模拟）

题目来源：

http://129.211.20.246/problem.php?id=1018

题目描述

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

   4+3x=8
   6a-5+1=2-2a
   -5+12y=0

ZL先生被主管告之，在计算器上键入的一个一元一次方程中，只包含整数、小写字母及+、-、=这三个数学符号（当然，符号"─"既可作减号，也可作负号）。方程中并没有括号，也没有除号，方程中的字母表示未知数。 
编写程序，解输入的一元一次方程，将解方程的结果（精确至小数点后三位）输出至屏幕。
你可假设对键入的方程的正确性的判断是由另一个程序员在做，或者说可认为键入的一元一次方程均为合法的，且有唯一实数解。 

输入

每个测试文件只包含一组测试数据，每组输入一个字符串，表示一个一元一次方程。 

输出

对于每组输入数据，解方程的结果（精确至小数点后三位）输出至屏幕。 

样例输入

6a-5+1=2-2a

样例输出 

a=0.750

来源/分类

2000年NOIP全国联赛普及组  

参考代码：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<iostream>
#define N 200005
#define mod 1000007
using namespace std;
char s[N];
int main()
{
    while(~scanf("%s",s))
    {
        double k1=0,b1=0,k2=0,b2=0,a=0;
        int flag=0,p=1;
        char x;
        for(int i=0; i<strlen(s); i++)
        {
            if(s[i]=='=')
            {
                p=1;
                b1+=a;
                a=0;
                for(int j=i+1; j<strlen(s); j++)
                {
                    if(s[j]=='+')
                    {
                        p=1;
                        b2+=a;
                        a=0;
                        continue;
                    }
                    else if(s[j]=='-')
                    {
                        p=-1;
                        b2+=a;
                        a=0;
                        continue;
                    }
                    if(s[j]>='0'&&s[j]<='9')
                    {
                        a*=10;
                        a+=(s[j]-'0')*p;
                    }
                    else
                    {
                        x=s[j];
                        if(a==0) a=1*p;
                        k2+=a;
                        a=0;
                    }
                    if(j==strlen(s)-1)
                        b2+=a;
                }
                break;
            }
            if(s[i]=='+')
            {
                p=1;
                b1+=a;
                a=0;
                continue;
            }
            else if(s[i]=='-')
            {
                p=-1;
                b1+=a;
                a=0;
                continue;
            }

            if(s[i]>='0'&&s[i]<='9')
            {
                a*=10;
                a+=(s[i]-'0')*p;
            }
            else
            {
                x=s[i];
                if(a==0) a=1*p;
                k1+=a;
                a=0;
            }
        }
        if((b2-b1)/(k1-k2)==0)
            printf("%c=%.3lf\n",x,fabs((b2-b1)/(k1-k2)));
        else
            printf("%c=%.3lf\n",x,(b2-b1)/(k1-k2));
    }
    return 0;
}