LeetCode-241. 为运算表达式设计优先级
给定一个含有数字和运算符的字符串,为表达式添加括号,改变其运算优先级以求出不同的结果。你需要给出所有可能的组合的结果。有效的运算符号包含 +, - 以及 * 。
示例 1:
输入: "2-1-1"
输出: [0, 2]
解释:
((2-1)-1) = 0
(2-(1-1)) = 2
示例 2:
输入: "2*3-4*5"
输出: [-34, -14, -10, -10, 10]
解释:
(2*(3-(4*5))) = -34
((2*3)-(4*5)) = -14
((2*(3-4))*5) = -10
(2*((3-4)*5)) = -10
(((2*3)-4)*5) = 10
思路
- 给一段带有运算符的字符串加上括号之后就会输出不同的结果。单纯地给一个数字加括号,如给2加成(2),2还是2。
- 我们从前向后遍历字符串,遇到运算符后就开始进行分而治之。分而治之的思想关键是,对于每个分治的部分,采取相同的计算逻辑。先计算这个运算符左边的子串加括号之后的各种结果,然后计算它右边子串加括号后的各种结果。
- 这种思想实际上是对这个运算符的左边和右边先加了一个大括号,然后对每个大括号中的子串施行一致的计算逻辑。直到遍历完这个子串。如果在遍历过程中,没有发现运算符,则这个大括号中只能存在一种计算结果,也就是单纯地给一个数字加括号的情况。
- 相关API
- substring(0,i) 是获得字符数组下标0到i-1处的子串。
- substring(i+1)是获得字符数组下标i+1到末尾的子串。具体重载方法请参照JDK。
class Solution {
public List<Integer> diffWaysToCompute(String input) {
List<Integer> res=new ArrayList<Integer>();
if(input==null) return res;
for(int i=0;i<input.length();i++)
{
char c=input.charAt(i);
if(c=='+' || c=='*' || c=='-')
{
//处理左边,收集左边的各种情况,此处的分治相当于给左边加了一个大括号
List<Integer> left=diffWaysToCompute(input.substring(0,i));
//处理右边,收集右边的各种情况,此处的分治相当于给右边加了一个大括号
List<Integer> right=diffWaysToCompute(input.substring(i+1));
//遍历左边和右边收集的加括号之后的各种计算结果
for(int l:left)
{
for(int r:right)
{
//然分治的结果是左右两边各加大括号的结果,所以将目光聚集在当前的运算符上即可
switch(c){
case '+':
res.add(l+r);
break;
case '-':
res.add(l-r);
break;
case '*':
res.add(l*r);
break;
}
}
}
}
}
//如果某次遍历完字符串没有发现运算符,那么直接将字符串转为int返回
//可以理解为给单独的一个数字加了一个无所谓有没有的括号
if(res.size()==0)
{
res.add(Integer.valueOf(input));
}
return res;
}
}