洛谷P1022 计算器的改良

P1022 计算器的改良

题目背景

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

题目描述

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

4＋3x＝8

6a－5＋1＝2－2a

－5＋12y＝0

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

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

输入输出格式

输入格式：

一个一元一次方程。

输出格式：

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

输入输出样例

输入样例#1：

6a-5+1=2-2a

输出样例#1：

a＝0.750

/*
    模拟移项
*/
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
char s[10010],ch;
double a,b;
int main(){
    freopen("Cola.txt","r",stdin);
    scanf("%s",s+1);
    int k1=1,k2=-1;
    for(int i=1;i<=strlen(s+1);){
        int w1=k1,w2=k2,x=0;
        bool flag=0;
        while(s[i]<'0'||s[i]>'9'){
            if(s[i]=='=')k1=-k1,k2=-k2,w1=k1,w2=k2;
            else if(s[i]=='-')flag=1;
            i++;
        }
        while(s[i]<='9'&&s[i]>='0')x=x*10+s[i]-'0',i++;
        if(flag)w1=-w1,w2=-w2;
        if(s[i]<='z'&&s[i]>='a')a+=x*w1,ch=s[i];
        else b+=x*w2;
    }
    double ans=b/a;
    if(ans==0)ans=0;
    printf("%c=%.3lf",ch,ans);
}