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;

}
}