题干:
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 *
.
Input: "2-1-1"
.
((2-1)-1) = 0
(2-(1-1)) = 2
Output: [0, 2]
这道题还是稍微有点难度。乍一看跟 282. Expression Add Operators 有一些相似,都是表达式添加一些东西最终变成一个值,但是解法却大相径庭。这道题用二分法而282用的是DFS。
在表达式加括号题干中,表达式是已经规定好的,也就是说每一个unit已经划分好了。每一个运算符号把整个表达式分成两部分,并且每一部分都是一个mini版的表达式加括号问题。如果一个表达式有n个unit,那么就有n-1种方法把它分成两部分,左右分别家一个括号,再用当前运算符号计算整个表达式的值。以此来遍历第一级的所有加括号的情况。对于左右两个表达式做同样处理。这种自顶向下迭代的过程是多么标准的二分法。。。
相比而言,对于282,expression Add Operators这道题,首先这些数字并未分片,所以如果只用加法链接,那么就是典型的从左至右切段段问题,和word break并没有什么区别,多么标准的DFS。不过这道题有乘法和减法的话就稍微加了一个细节,就是每一次切断的时候不仅要记录整个表达式的值,还要记录上个unit的值,因为也许和第二个unit运算(比如相乘)的时候需要用到。这些细节放到后面做这道题的时候在讨论。
那么就开始写这道题吧:
class Solution {
public:
vector<int> diffWaysToCompute(string input) {
vector<int> left;
vector<int> right;
vector<int> res;
int num=0;
for(int i=0;i<input.size();++i){
if(input[i]>='0'&&input[i]<='9')
continue;
left = diffWaysToCompute(input.substr(0,i));
right = diffWaysToCompute(input.substr(i+1));
for(auto x:left){
for(auto y:right){
if(input[i]=='+'){
res.push_back(x+y);
}
else if(input[i]=='-'){
res.push_back(x-y);
}
else{
res.push_back(x*y);
}
}
}
}
if(res.empty())
res.push_back(stoi(input));
return res;
}
};
这个做法简单易懂,lc也能ac,用时8ms。不过貌似还有一些优化的空间,因为有很多地方都重复计算了。对于一个表达式m和m+1处cut时,对于n<m,1~n的这个区间就会被重复计算。可以用一个hashmap记录中间结果来避免重复计算。好不想写code啊好想睡觉。。。。。但是再懒得话就会gg第二次。。。哭哭。。。
要注意的是怎么处理unordered_map的key,以及recursion 调用helper的时候begin和end的值
class Solution {
public:
unordered_map<string, vector<int>> memo;
vector<int> diffWaysToCompute(string input) {
vector<int> res=helper(input, 0, input.size());
return res;
}
string assem_key(int start, int end){
stringstream ss;
ss<<start<<'-'<<end;
return ss.str();
}
vector<int> helper(string input, int begin, int end){
string key=assem_key(begin, end);
if(memo.find(key)!=memo.end())
return memo[key];
vector<int> left;
vector<int> right;
vector<int> res;
int num=0;
for(int i=0;i<input.size();++i){
if(input[i]>='0'&&input[i]<='9')
continue;
left = helper(input.substr(0,i), begin, begin+i-1);
right = helper(input.substr(i+1), begin+i+1, end);
for(auto x:left){
for(auto y:right){
if(input[i]=='+'){
res.push_back(x+y);
}
else if(input[i]=='-'){
res.push_back(x-y);
}
else{
res.push_back(x*y);
}
}
}
}
if(res.empty())
res.push_back(stoi(input));
memo[key]=res;
return res;
}
};
哎我插入code的格式为什么一直有错。。