题目地址:
https://www.lintcode.com/problem/convert-expression-to-polish-notation/description
给定一个中缀表达式,要求将其转为前缀表达式。
方法是从右向左遍历表达式,然后运用中缀转后缀的方法来做,思路可以参考https://blog.csdn.net/qq_46105170/article/details/106207612。具体方法如下:
1、开一个栈存符号,开一个列表存答案;
2、遇到右括号则入栈,遇到数字则add到列表中;
3、如果遇到左括号,则不停将栈顶pop出来直接append到列表后面;
4、如果遇到运算符,则看一下栈顶的运算符和遇到的运算符优先级如何,如果比之严格大,则栈顶出栈,直接append到列表后面,直到栈顶的优先级不大于新来的运算符时,再将新来的运算符入栈;
5、遍历完之后,从栈顶到栈底,依次将元素add到列表后面。
6、最后将列表反序。
注意,这里与中缀转后缀有一个细节是不同的,中缀转后缀的时候,遇到运算符,是栈顶优先级只需要大于等于新来的运算符就能出栈add到答案里了,但是这里需要严格大于,原因是我们是从右向左算的,需要遇到严格大于的运算符才能知道栈顶的运算符应该计算了。代码如下:
import java.util.*;
public class Solution {
/**
* @param expression: A string array
* @return: The Polish notation of this expression
*/
public List<String> convertToPN(String[] expression) {
// write your code here
List<String> res = new ArrayList<>();
Deque<String> ops = new LinkedList<>();
Map<String, Integer> prio = new HashMap<>();
prio.put(")", 1);
prio.put("+", 2);
prio.put("-", 2);
prio.put("*", 3);
prio.put("/", 3);
for (int i = expression.length - 1; i >= 0; i--) {
String s = expression[i];
if (Character.isDigit(s.charAt(0))) {
res.add(s);
} else if (s.equals(")")) {
ops.push(s);
} else if (s.equals("(")) {
while (!ops.peek().equals(")")) {
res.add(ops.pop());
}
ops.pop();
} else {
while (!ops.isEmpty() && prio.get(ops.peek()) > prio.get(s)) {
res.add(ops.pop());
}
ops.push(s);
}
}
res.addAll(ops);
Collections.reverse(res);
return res;
}
}
时空复杂度 O ( n ) O(n) O(n)。