【LeetCode】241. 为运算表达式设计优先级 结题报告 (C++)

原题地址：https://leetcode-cn.com/problems/different-ways-to-add-parentheses/

题目描述：

给定一个含有数字和运算符的字符串，为表达式添加括号，改变其运算优先级以求出不同的结果。你需要给出所有可能的组合的结果。有效的运算符号包含 +, - 以及 * 。

示例 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*3-4*5】 
(2)*(3-4*5)、(2*3)-(4*5)、(2*3-4)*5

刚开始读题时，对答案的个数存在着疑问。以为共有3个符号，所以共有3*2*1=6种答案。这是对题目理解不够，其实只有五个答案。五个答案怎么来的，是根据分治算法的思想得到的。

代码：

class Solution {
public:
    int cal(char e,int a,int b){
        switch(e){
            case '+': return a + b;
            case '-': return a - b;
            case '*': return a * b;
            case '/': return a / b;
            default: return 0;
        }
    }
    vector<int> diffWaysToCompute(string input) {
        vector<int> ans, tmp1, tmp2;
        if(input.length() == 0)
            return ans;
        int flag = 0;
        for(int i = 0; i < input.length(); i++){
            switch(input[i]){
                case '+':
                case '-':
                case '*':
                case '/':
                    flag = 1;
                    break;
            }
        }
        if(flag == 0) 
            return vector<int>(1,stoi(input));
        
        for(int i = 0; i < input.length(); i++){
            switch(input[i]){
                case '+':
                case '-':
                case '*':
                case '/':
                    tmp1 = diffWaysToCompute(input.substr(0,i));
                    tmp2 = diffWaysToCompute(input.substr(i+1,input.length()));
                    for(auto& x1:tmp1)
                        for(auto& x2:tmp2)
                            ans.push_back(cal(input[i],x1,x2));
                    break;
                default:
                    break;
            }
        }
        
        return ans;    
    }
};