力扣241-为运算表达式设计优先级-C++分治算法

一、题目

给你一个由数字和运算符组成的字符串 expression ，按不同优先级组合数字和运算符，计算并返回所有可能组合的结果。你可以 按任意顺序 返回答案。

示例 1：

输入：expression = "2-1-1"
输出：[0,2]
解释：
((2-1)-1) = 0 
(2-(1-1)) = 2
示例 2：

输入：expression = "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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/different-ways-to-add-parentheses

二、思路

这里的思路参考：https://leetcode-cn.com/problems/different-ways-to-add-parentheses/solution/cfen-zhi-by-heroding-3v1c/

对于表达式中的每个运算符，都进行一次划分，先分别计算其左右两边的结果，然后再将左右两边的结果进行组合（根据当前运算符所表示的运算进行组合）。 这里仅记录作为学习分治使用。

三、代码

class Solution {
public:
    vector<int> diffWaysToCompute(string expression) {
        vector<int> ans;
        for(int i=0; i<expression.size(); i++){
            char ch = expression[i];
            // 当前字符是运算符
            if(ch == '+' || ch == '-' || ch == '*'){
                // 运算符左边的计算结果  分治 & 递归
                vector<int> left = diffWaysToCompute(expression.substr(0, i));
                // 运算符右边的计算结果
                vector<int> right = diffWaysToCompute(expression.substr(i+1));
                // 将左右两边的计算结果组合
                for(int& l : left){
                    for(int& r : right){
                        switch(ch){
                            case '+':
                                ans.push_back(l + r);
                                break;
                            case '-':
                                ans.push_back(l - r);
                                break;
                            case '*':
                                ans.push_back(l * r);
                                break;
                        }
                    }
                }
            }
        }
        // expression中没有运算符，只有一个操作数
        if(ans.size() == 0){
            // atoi(expression.c_str())
            ans.push_back(stoi(expression));
        }
        return ans;
    }
};