1. 程式人生 > >leetcode之為運算表示式設計優先順序

leetcode之為運算表示式設計優先順序

leetcode之為運算表示式設計優先順序

給定一個含有數字和運算子的字串,為表示式新增括號,改變其運算優先順序以求出不同的結果。你需要給出所有可能的組合的結果。有效的運算子號包含 +, - 以及 * 。

示例 1:

輸入: “2-1-1”
輸出: [0, 2]
解釋:
((2-1)-1) = 0
(2-(1-1)) = 2
示例 2:

輸入: “23-45”
輸出: [-34, -14, -10, -10, 10]
解釋:
(2*(3-(45))) = -34
((2
3)-(45)) = -14
((2
(3-4))5) = -10
(2
((3-4)5)) = -10
(((2
3)-4)*5) = 10

思路:

我發現我們可以分成一小段一小段來解決,對於每一個運算子,都可以作為最後一割運算,對左右兩邊的結果進行操作。如果要獲得左右兩邊的結果,我們可以把他們當作一個新的引數作為輸入,這便是用到了遞迴的方法。如果每一個字串中沒有操作符,說明是單個數字,需要直接存入到集合中。

class Solution {
    public List<Integer> diffWaysToCompute(String input) {
        List<Integer> res=new ArrayList<Integer>();
        for(int i=0;i<input.length();i++){
            char c=input.charAt(i);
            if(c=='*'||c=='+'||c=='-'){
                List<Integer> leftList=diffWaysToCompute(input.substring(0,i));
                List<Integer> rightList=diffWaysToCompute(input.substring(i+1));
                if(c=='*'){
                    for(int left:leftList){
                        for(int right:rightList){
                            res.add(left*right);
                        }
                    }
                }else if(c=='-'){
                     for(int left:leftList){
                        for(int right:rightList){
                            res.add(left-right);
                        }
                    }
                }else if(c=='+'){
                     for(int left:leftList){
                        for(int right:rightList){
                            res.add(left+right);
                        }
                    }
                }
            }
        }
        if(res.isEmpty()){
            res.add(Integer.parseInt(input));
        }
        return res;
        
    }
}