给定一个含有数字和运算符的字符串,为表达式添加括号,改变其运算优先级以求出不同的结果。你需要给出所有可能的组合的结果。有效的运算符号包含 +, - 以及 * 。
示例 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
思路:本题我们采用分治法。
写法一:
class Solution {
ArrayList<Character> op=new ArrayList();
String[] nums;
public List<Integer> diffWaysToCompute(String input) {
split(input);
return patition(0,nums.length-1);
}
private void split(String input)
{
for(int i=0;i<input.length();i++)
{
if(input.charAt(i)=='+' || input.charAt(i)=='-'
|| input.charAt(i)=='*')
op.add(input.charAt(i));
}
nums=input.split("[\\+\\-\\*]");
}
private List<Integer> patition(int start,int end)
{
List<Integer> result=new ArrayList();
if(start==end)
{
result.add(Integer.valueOf(nums[start]));
return result;
}
for(int i=start;i<end;i++)
{
for(Integer left:patition(start,i))
for(Integer right:patition(i+1,end))
{
if(op.get(i)=='+')
result.add(left+right);
else if(op.get(i)=='-')
result.add(left-right);
else if(op.get(i)=='*')
result.add(left*right);
}
}
return result;
}
}
写法二:
class Solution {
public List<Integer> diffWaysToCompute(String input) {
return patition(input);
}
private List<Integer> patition(String input)
{
List<Integer> result=new ArrayList();
if(!input.contains("+") && !input.contains("-") && !input.contains("*"))
{
result.add(Integer.valueOf(input));
return result;
}
for(int i=0;i<input.length();i++)
{
if(input.charAt(i)=='+' || input.charAt(i)=='-'
|| input.charAt(i)=='*')
{
for(Integer left:patition(input.substring(0,i)))
for(Integer right:patition(input.substring(i+1)))
{
if(input.charAt(i)=='+')
result.add(left+right);
else if(input.charAt(i)=='-')
result.add(left-right);
else if(input.charAt(i)=='*')
result.add(left*right);
}
}
}
return result;
}
}