[Leetcode] Different Ways to Add Parentheses

Different Ways to Add Parentheses

Given a string of numbers and operators, return all possible results from computing all the different possible ways to group numbers and operators. The valid operators are +, - and *.

Example 1

Input: "2-1-1".

((2-1)-1) = 0
(2-(1-1)) = 2

Output: [0, 2]

这个题目刚开始拿到的时首先想到的是这样的思路：产生所有的括号组合方式，也就是产生所有的expression，然后调用计算表达式的方法求取结果。

后来发现，题目的提示使用分治算法，OMG！！

一、首先将input字符串分解为两部分

data和operator

比如2*3-4*5

data：      2 | 3 | 4 | 5

operator：* | - | *

我们可以不断的选择中点mid将data划分为两部分，这里mid=[0,1,2]

分别表示划分的left部分的最后一个数字，对应的right部分的开始是从[1,2,3]

实际上思路已经很明了，直接贴一下代码

import java.util.*;

public class Solution {
    private List<Integer> divide(Vector<Character> op, Vector<Integer> data, int dstart,int dend){
        if(dstart==dend){
            List<Integer> res = new LinkedList<Integer>();
            res.add(data.get(dstart));
            return res;
        }
        List<Integer> res = new LinkedList<Integer>();
        for(int mid = dstart;mid<dend;mid++){
            List<Integer> left = divide(op,data,dstart,mid);
            List<Integer> right = divide(op,data,mid+1,dend);
            char opc = op.get(mid);
            for(int i=0;i<left.size();i++){
                for(int j=0;j<right.size();j++){
                    int tmp = 0;
                    switch(opc){
                        case '+':
                            tmp = left.get(i)+right.get(j);
                        break;
                        case '-':
                            tmp = left.get(i)-right.get(j);
                        break;
                        case '*':
                            tmp = left.get(i)*right.get(j);
                        break;
                    }
                    res.add(tmp);
                }
            }
        }
        return res;
    }
    public List<Integer> diffWaysToCompute(String input) {
        Vector<Character> op = new Vector<Character>();
        Vector<Integer> data = new Vector<Integer>();
        int index = 0;
        while(index<input.length()){
            char c = input.charAt(index);
            if(c=='+'||c=='-'||c=='*'){
                op.add(c);
                index++;
            }else{
                int endindex = index+1;
                while(endindex<input.length()&&Character.isDigit(input.charAt(endindex)))endindex++;
                String tmp = input.substring(index,endindex);
                data.add(Integer.parseInt(tmp));
                index = endindex;
            }
        }
        return divide(op,data,0,data.size()-1);
    }
}