http://www.lintcode.com/en/problem/convert-expression-to-reverse-polish-notation/
中缀表达式转成后缀表达式
For the expression [3 - 4 + 5]
(which denote by ["3", "-", "4", "+", "5"]), return [3 4 - 5 +]
(which denote by ["3", "4", "-", "5", "+"])
中缀表达式a + b*c + (d * e + f) * g,其转换成后缀表达式则为a b c * + d e * f + g * +。
转换过程需要用到栈,具体过程如下:
1)如果遇到操作数,我们就直接将其输出。
2)如果遇到操作符,则我们将其放入到栈中,遇到左括号时我们也将其放入栈中。
3)如果遇到一个右括号,则将栈元素弹出,将弹出的操作符输出直到遇到左括号为止。注意,左括号只弹出并不输出。
4)如果遇到任何其他的操作符,如(“+”, “*”,“(”)等,从栈中弹出元素直到遇到发现更低优先级的元素(或者栈为空)为止。弹出完这些元素后,才将遇到的操作符压入到栈中。有一点需要注意,只有在遇到" ) "的情况下我们才弹出" ( ",其他情况我们都不会弹出" ( "。
5)如果我们读到了输入的末尾,则将栈中所有元素依次弹出。
public class Solution {
/**
* @param expression: A string array
* @return: The Reverse Polish notation of this expression
*/
public ArrayList<String> convertToRPN(String[] expression) {
// write your code here
ArrayList<String> res = new ArrayList();
Stack<String> stack = new Stack();
HashMap<String, Integer> map = new HashMap();
// 优先级越高值越大
map.put("+", 1);
map.put("-", 1);
map.put("*", 2);
map.put("/", 2);
map.put("(", 3);
for (String s : expression) {
if (Character.isDigit(s.charAt(0))) {
res.add(s);
} else {
if (s.equals(")")) {
while (!stack.isEmpty() && !stack.peek().equals("(")) {
res.add(stack.pop());
}
if (!stack.isEmpty()) {
stack.pop();
}
} else {
while (!stack.isEmpty() && map.get(s) <= map.get(stack.peek())) {
// “(”只有遇到右括号才出栈,否则break
if (stack.peek().equals("(")) {
break;
}
res.add(stack.pop());
}
stack.push(s);
}
}
}
while (!stack.isEmpty()) {
res.add(stack.pop());
}
return res;
}
}