读算法书,发现书上的 Dijkstra 双栈算法运行不了,原因在于 java 的 scanner 类有时候很难终止输入,程序不能正确运行。
于是改写了原程序,让程序并不从控制台通过 scanner 类读写表达式,而是在程序里用字符串定义好表达式,就能正确运行了。
该算法的原理是:通过两个栈,一个存符号,一个存数字,然后通过右括号来控制栈的进出与运算操作。
改写后的代码如下:
import java.util.Stack;
import edu.princeton.cs.algs4.StdOut;
public class Evaluate2 {
public static void main(String[] args) {
Stack<String> ops = new Stack<String>();
Stack<Double> vals = new Stack<Double>();
String str = "( 1 + ( ( 2 + 3 ) * ( 4 * 5 ) ) )"; // 改写
String[] strs = str.split("\\s+");
for (String s : strs) {
// Read token, push if operator.
if (s.equals("(")) ;
else if (s.equals("+")) ops.push(s);
else if (s.equals("-")) ops.push(s);
else if (s.equals("*")) ops.push(s);
else if (s.equals("/")) ops.push(s);
else if (s.equals("sqrt")) ops.push(s);
else if (s.equals(")"))
{ // Pop, evaluate, and push result if token is ")".
String op = ops.pop();
double v = vals.pop();
if (op.equals("+")) v = vals.pop() + v;
else if (op.equals("-")) v = vals.pop() - v;
else if (op.equals("*")) v = vals.pop() * v;
else if (op.equals("/")) v = vals.pop() / v;
else if (op.equals("sqrt")) v = Math.sqrt(v);
vals.push(v);
} // Token not operator or paren: push double value.
else vals.push(Double.parseDouble(s));
}
StdOut.println(vals.pop());
}
}