241. 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"
Output: [0, 2]
Explanation: 
((2-1)-1) = 0 
(2-(1-1)) = 2
Example 2:

Input: "2*3-4*5"
Output: [-34, -14, -10, -10, 10]
Explanation: 
(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

这题明显是一个divide and conquer题,就是要递归分治。但这里有两个递归的思路,一个是跟着数字走,四个数字变成三个数字,再变成两个数字,以此为基础计算,输出结果大概是对的,但会有重复,比如题干里的第二个例子,2和3先结合,或者4和5先结合,会有重复的考虑,不太好避免。所以第二个划分方法更好,就是从运算符号入手,毕竟,针对所有结果的讨论可以归结到最后是进行哪个符合运算结束的。
代码如下:

class Solution {
public:
    vector<int> diffWaysToCompute(string input) {
        vector<int> res;
        for (int i = 0; i < input.size(); i++) {
            char ch = input[i];
            if (ch == '+' || ch == '-' || ch == '*') {
                vector<int> left = diffWaysToCompute(input.substr(0, i));
                vector<int> right = diffWaysToCompute(input.substr(i + 1, input.size() - i - 1));
                for (int j = 0; j < left.size(); j++) {
                    for (int k = 0; k < right.size(); k++) {
                        if (ch == '+')
                            res.push_back(left[j] + right[k]);
                        else if (ch == '-')
                            res.push_back(left[j] - right[k]);
                        else if (ch == '*')
                            res.push_back(left[j] * right[k]);
                    }
                }
            }
        }
        if (res.size() == 0) res.push_back(stoi(input));
        return res;
    }
};

优化,这里很类似rod cut 那个DP的引入例子,递归可以做,但是会有重复的子问题被多次计算,所以可以试着把计算过的结果存起来。这里引入一个外部的hash table。如下:

class Solution {
private:
    unordered_map<string, vector<int>> hash;
public:
    vector<int> diffWaysToCompute(string input) {
        vector<int> res;
        if (hash.find(input) != hash.end()) return hash[input];
        for (int i = 0; i < input.size(); i++) {
            char ch = input[i];
            if (ch == '+' || ch == '-' || ch == '*') {
                vector<int> left = diffWaysToCompute(input.substr(0, i));
                vector<int> right = diffWaysToCompute(input.substr(i + 1, input.size() - i - 1));
                for (int j = 0; j < left.size(); j++) {
                    for (int k = 0; k < right.size(); k++) {
                        if (ch == '+')
                            res.push_back(left[j] + right[k]);
                        else if (ch == '-')
                            res.push_back(left[j] - right[k]);
                        else if (ch == '*')
                            res.push_back(left[j] * right[k]);
                    }
                }
            }
        }
        if (res.size() == 0) res.push_back(stoi(input));
        hash[input] = res;
        return res;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值