codevs.1015 计算器的改良

题目描述 Description

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

 

   4＋3x＝8

   6a－5＋1＝2－2a

　 －5＋12y＝0

 

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

 

编写程序，解输入的一元一次方程, 将解方程的结果(精确至小数点后三位)输出至屏幕。

 

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

输入描述 Input Description

一个一元一次方程

输出描述 Output Description

方程的解

样例输入 Sample Input

6a-5+1=2-2a

样例输出 Sample Output

a=0.750

数据范围及提示 Data Size & Hint

注意加减号都是半角

一开始觉得好难啊，后面发现模拟计算等号左右的系数和常数，发现也可以做

#include<cstdio>
#include<iostream>
#include<string>
#include<algorithm>
#include<iomanip>
using namespace std;
const int maxn = 5;
int pos,n[maxn],u[maxn],lr=0,k;
double l=0,r=0;
string s;
void cal(bool sign);
int main(){
   freopen("Text.txt","r",stdin);
    while(cin>>s){
        k=0;
        while(k<s.size()){
            if(s[k]=='-'){
                k++;
                cal(1);
                
            }
            else if(s[k]=='+'){
                k++;
                cal(0);
                
            }
            else{
                if(s[k]=='='){
                    k++;
                    lr=1;
                    continue;
                }
                cal(0);
            }
        }
        l=n[0]-n[1];
        r=u[1]-u[0];
        //cout<<n[0]<<" "<<n[1]<<endl;
        //cout<<l<<" "<<r<<endl;
        cout.setf(ios::fixed);
        double ans=l/r;
        if(ans>0)ans+=0.0005;
        else ans -= 0.0005;
        cout<<setprecision(3)<<s[pos]<<" = "<< ans <<endl;
        //printf("%.3lf",l/r+0.0005);
    }
   return 0;
}
void cal(bool sign){
    //cout<<"k = "<<k<<endl;
    //cout<<s[k]<<endl;
   bool flag=0;
   int tal=0;
   while(s[k]>='0'&&s[k]<='9'){
       flag=1;
       tal=tal*10+s[k]-'0';
       k++;
   }
   if(sign)tal=-tal;
   if(!flag)tal=1;
   if(s[k]>='a'&&s[k]<='z'){
       pos=k;
       u[lr]+=tal;
       k++;
   }
   else{
      n[lr]+=tal;
      
   }
}