题目描述:
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
思路:这道题的本质就是树的构建(Unique Binary Search Trees II扩展),难点在于想到利用运算符号作为根结点来划分左右子树,依次选取所传入字符串的各个运算符号作为根节点划分左右子树,同时对左右子树做相同的操作,最后即可得到结果。要注意的是函数每一次递归返回的是一个vector里面存储的是根节点下可以形成的所有形式左右子树运算结果的集合。我们对题目解法进行了一次优化,即用一个哈希表来保存传入字串所能得到的所有运算结果,当下面某一次划分子树过程中再次遇见直接返回原来所求得的结果,而不用再进行划分求解的过程。
下面用一些示意图来表示会更加容易理解:
#include <cstdio>
#include<vector>
#include<string>
#include<map>
#include <stdlib.h>
using namespace std;
map<string,vector<int> > record;
int Calculate(int num1,char operation,int num2)
{
int result;
if(operation=='+')
result = num1+num2;
else
if(operation=='-')
result = num1-num2;
else
result = num1*num2;
return result;
}
vector<int> diffWaysToCompute(string input)
{
vector<int> result;
//之前已经遇到过一次相同的子树那么直接返回之前得到的vector即可
if(record.find(input) != record.end())
return record[input];
for(int i=0;i<input.size();i++)
{
//遇到运算符号就进行左右子树划分
if(input[i]=='+'||input[i]=='-'||input[i]=='*')
{
//左子树
vector<int> num1 = diffWaysToCompute(input.substr(0,i));
//右子树
vector<int> num2 = diffWaysToCompute(input.substr(i+1));
//子树根节点下左右子树的各种情况的组合
for(int m=0;m<num1.size();m++)
for(int n=0;n<num2.size();n++)
result.push_back(Calculate(num1[m],input[i],num2[n]));
}
}
//对叶子结点或者一开始就没有预算符的input处理
if(result.size()==0)
result.push_back(atoi(input.c_str()));
//对每一次input的构成的所有子树情况进行记录
record[input] = result;
return result;
}
int main(){
string input = "11";
vector<int> result = diffWaysToCompute(input);
printf("结果个数为%d\n",result.size());
for(int i=0;i<result.size();i++)
printf("%d\n",result.at(i));
return 0;
}