TOJ 4829: 计算器的改良

Python写法！

传送门：http://acm.tzc.edu.cn/acmhome/problemdetail.do?&method=showdetail&id=4829

描述

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

4+3x+8

6a-5+1=2-2a

　 -5+12y=0

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

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

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

输入

一个一元一次方程（见上，长度不超过100）

输出

方程的解,精确到小数点后三位.

样例输入

6a-5+1=2-2a

样例输出

a=0.750

 

思路：

　def solve(eq,var):
    eq1 = eq.replace("=","-(")+")"
    c = eval(eq1,{var:1j})
    return -c.real/c.imag

主要是用上面的函数求解了一元一次方程，参数是方程，以及未知量，因为要使用上面的函数还得把题目中的未知量，如果未知量前面有非1系数的话，得加上*号，其次是用re.findall("[a-z]+",str) 提取字符串里面的未知量

代码：

import re
def solve(eq,var):
    eq1 = eq.replace("=","-(")+")"
    c = eval(eq1,{var:1j})
    return -c.real/c.imag
if __name__ =="__main__":
    while True:
        try:
            str = input()
            s = re.findall("[a-z]+",str)
            str = str.replace(' ','')
            flag = 0
            while True:
                flag = 1
                for i in range(1,len(str)):
                    if(str[i] == s[0] and str[i-1]!='=' and str[i-1]!='+'and str[i-1]!='-' and str[i-1]!='*'):
                        str = str[:i]+'*'+str[i:]
                        flag = 0
                        break
                if(flag == 1):break
            print("%s=%.3f"%(s[0],solve(str,s[0])))
        except EOFError:
            break