题目描述:
(改题引自力扣第241题)
给定一个含有数字和运算符的字符串,为表达式添加括号,改变其运算优先级以求出不同的结果。你需要给出所有可能的组合的结果。有效的运算符号包含 +, - 以及 * 。
输入
输入为一行字符串
输出
输出为所有可能的结果,每个一行,从小到大排列
输入样例1:
2-1-1
输出样例1:
0
2
输入样例2:
2*3-4*5
输出样例2:
-34
-14
-10
-10
10
题目分析:
这题使用的是分治算法,把大问题分解成小问题,先求小再求大。按照递归的运算顺序,作为主串分界点的运算符优先级最低,即最后作运算。递归的出口为串中不含运算符。
对于运算表达式A和B组成的表达式A+B,如果把里面的+作为最低优先级的符号,则该问题可以先分别求子问题A和B的最终值,再将A的每一个最终值与B的每一个最终值做+运算,便可以得到A+B的最终值。
对于任意一个运算表达式S,可以以S中的每一个运算符分别作为最低优先级的符号,先对两侧的字符串进行递归求解,再对两个得到的数组做运算。遍历完所有运算符后,得到的数组便是S的所有最终值。
当S是一个数字串时,我们使用stoi()函数将其转为整数存入数组中,然后返回数组即可。
解决方案:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int cal(int n1,int n2,char ch){ //用于两数作运算
if(ch=='+')
return n1+n2;
if(ch=='-')
return n1-n2;
if(ch=='*')
return n1*n2;
return n1/n2;
}
bool IAN(string str){ //判断是否为数字串
for(int i=0;i<str.size();i++)
if(str[i]<'0'||str[i]>'9')
return false;
return true;
}
vector<int> calculate(string chs){
vector<int>p;
if(IAN(chs)){ //数字串,递归出口
p.push_back(stoi(chs));
return p;
}
for(int i=0;i<chs.size();i++){
if(chs[i]<'0'){ //该字符作为该串的最低优先级运算符
vector<int>p1=calculate(chs.substr(0,i)),p2=calculate(chs.substr(i+1,chs.size()-1-i)); //子问题求解,左右两侧子串的运算结果
for(int j=0;j<p1.size();j++)
for(int k=0;k<p2.size();k++)
p.push_back(cal(p1[j],p2[k],chs[i])); //子问题归并
}
}
return p;
}
int main(){
string str;
cin>>str;
vector<int>p=calculate(str);
sort(p.begin(),p.end()); //排序
for(int i=0;i<p.size();i++)
cout<<p[i]<<endl;
return 0;
}