241. 为运算表达式设计优先级
给定一个含有数字和运算符的字符串,为表达式添加括号,改变其运算优先级以求出不同的结果。你需要给出所有可能的组合的结果。有效的运算符号包含 +, - 以及 * 。
示例 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
解题
需要对括号的所有位置进行遍历;
若使用回溯,需要返回每次放了括号的结果,条件难以设置;
分治+递归
采用分治,每次取符号位置,得到符号左边的式子的所有结果,与右边式子的所有排列结果,二重循环的排列计算,即可得到当前结果;
注意点
使用递归时明确递归的结果!
(1)递归返回的为当前字符串的公式结果列表
(2)最小条件:若当前字符串无符号,则返回字符串存储的数字
class Solution {
public:
vector<int> diffWaysToCompute(string input) {
//找到符号,对符号左右进行递归求解,然后得到左右的列表,进行排列组合的计算
vector<int> res;
int n=input.size();
for(int i=0;i<n;i++){
char c=input[i];
if(c=='+'||c=='-'||c=='*'){
//当遍历到符号时,取符号左右的所有计算结果
auto res1=diffWaysToCompute(input.substr(0,i));
auto res2=diffWaysToCompute(input.substr(i+1));
//开始对左右结果列表进行排列组合
for(auto r1:res1)
for(auto r2:res2)
if(c=='+') res.push_back(r1+r2);
else if(c=='-') res.push_back(r1-r2);
else if(c=='*') res.push_back(r1*r2);
}
}
//分治递归的最小单元——只剩下一个数——即上述遍历中无符号,故res无数据加入
if(res.empty()) res.push_back(stoi(input));
return res;
}
};