leetcode#241 为运算表达式设计优先级
题目:
给定一个含有数字和运算符的字符串,为表达式添加括号,改变其运算优先级以求出不同的结果。你需要给出所有可能的组合的结果。有效的运算符号包含 +, - 以及 * 。
示例:
输入: "2-1-1"
输出: [0, 2]
解释:
((2-1)-1) = 0
(2-(1-1)) = 2
思路:
对于一个形如 x op y(op 为运算符,x 和 y 为数) 的算式而言,它的结果组合取决于 x 和 y 的结果组合数,而 x 和 y 又可以写成形如 x op y 的算式。
因此,该问题的子问题就是 x op y 中的 x 和 y:以运算符分隔的左右两侧算式解。
然后我们来进行 分治算法三步走:
- 分解:按运算符分成左右两部分,分别求解
- 解决:实现一个递归函数,输入算式,返回算式解
- 合并:根据运算符合并左右两部分的解,得出最终解
代码:
class Solution
{
public:
vector<int> diffWaysToCompute(string expression)
{
int n = expression.length();
vector<int> tmp;
for (int i = 0; i < n; ++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, n - i - 1));
for (auto &i : left)
for (auto &j : right)
switch (ch)
{
case ('+'):
tmp.push_back(i + j);
break;
case ('-'):
tmp.push_back(i - j);
break;
case ('*'):
tmp.push_back(i * j);
break;
}
}
}
if (!tmp.size())
tmp.push_back(stoi(expression));
return tmp;
}
};