问题
编写一道程序,从标准输入得到一个缺少左括号的表达式并打印出补全括号之后的中序表达式。例如,给定输入:
1 + 2 ) * 3 - 4 ) * 5 - 6 ) ) )
你的程序应该输出:
((1 + 2) * ((3 - 4) * (5 - 6)))
思路
可以采用 Dijkstra 发明的“双栈法”:
- 将操作数压入操作数栈;
- 将运算符压入运算符栈;
- 忽略左括号;
- 在遇到右括号时,弹出一个运算符,弹出所需数量的操作数,并将预算结果压入操作数栈。
摘自《算法(第四版)》 P128
双栈法分析:
- 将操作数和运算符分类,压入不同的栈;
- 忽略左括号;
- 将右括号作为一个标志,每当遇到右括号,就弹出数据,进行处理;
- 这里的处理可以是对数值进行计算,也可以是其他的处理方法,比如本题中对字符串进行链接处理,之后,将处理结果压入操作数栈;
本题思路:
- 用空格将输入的数字和符号分隔开,每次读取一个字符串;
- 遇到数字就压入操作数栈,遇到运算符就压入运算符栈;
- 遇到右括号,就弹出两个操作数和一个运算符,进行字符串链接,而后作为一个字符串(整体)重新压入操作数栈;
- 由于栈中内容是反的,所以在字符串链接以及最后输出结果时都要额外处理一下,具体见代码。
代码:
package execise.chapter1_3;
import edu.princeton.cs.algs4.Stack;
import edu.princeton.cs.algs4.StdIn;
import edu.princeton.cs.algs4.StdOut;
/**
* Created by Blue on 2016/7/3.
* 从标准输入得到一个缺少左括号的表达式并打印出补全括号之后的中序表达式;
* 输入: 1 + 2 ) * 3 - 4 ) * 5 - 6 ) ) )
* 输出: (( 1 + 2 ) * (( 3 - 4 ) * ( 5 - 6 ) ) )
*/
public class E9 {
public static void main(String[] args) {
Stack<String> ops = new Stack<>(); // 符号栈
Stack<String> vals = new Stack<>(); // 数值栈,由于不需要运算,参数类型使用String,而非double;
while (!StdIn.isEmpty()) {
String s = StdIn.readString(); // 读取字符串,每次一个,以空格隔开;
if (s.equals("+") || s.equals("-") || s.equals("*") || s.equals("/"))
ops.push(s);
else if (s.equals(")")) {
// 此处,两边的括号是“)”、“(”,而非“(”、“)”;
String expre = ")" + vals.pop() + ops.pop() + vals.pop() + "(";
// 注意,整个表达式作为一个整体,也就是单个字符串压入栈顶;
vals.push(expre);
} else
vals.push(s);
}
String result = vals.pop();
// output: result: )))6-5(*)4-3((*)2+1((
StdOut.println("result: " + result);
// 更改逆序为顺序
String seq = "";
for (int i = result.length()-1; i >= 0; i--){
seq += result.charAt(i);
}
// output: seq: ((1+2)*((3-4)*(5-6)))
StdOut.println("seq: " + seq);
}
}
参考链接
小一的专栏