1096. 花括号展开 II
思路与分析
题目描述里其实有暗示,看到花括号和集合运算,第一时间想到了栈,但是CPU烧了,不会,以前学数据结构的时候做四则运算就不是自己做的。印象中的大概方法是,遇到乘除或者括号就先运算,遇到加减就往栈里存。但是这个题完全没思路,因为要的也不是单一的计算结果。
然后想到可以DFS递归处理每个花括号的内容,如果只有加法的话很好搞,但是乘法有一点点麻烦。犹豫再三还是点进了题解区,逛了之后做了DFS的方法。栈的方法就不赘述了,官解的说法非常清晰还有图解,具体看这里。
DFS
其实DFS是最容易想到的办法了。题目看起来复杂,其实简化下来就是,逗号表示或,逗号之间的元素任选一;挨着表示乘,乘子是字母的话直接拼接上去,是括号的话,括号里选一个乘。所以我们每次都处理最里面一层的花括号,如果有逗号,那把逗号里面的元素每一个单独拿出来和这个花括号以外的字符串拼接,产生子递归;如果没逗号,那直接把这层花括号去掉,与两端的拼接起来,只产生一个递归分支。具体看图,分别画了两种情况的图解。图片参考、代码参考。
class Solution {
public:
vector<string> braceExpansionII(string expression) {
map<string,int> vis;//哈希表,去重并自动排序
vector<string> res;
function<void(string)> dfs=[&](string s){
int j=s.find_first_of('}');//从左到右找第一个右括号
if(j==string::npos){
// cout<<s<<endl;
vis[s]++;
return;
}
int i=s.rfind('{',j);//逆向查找与其对应的左括号
string left=s.substr(0,i);//左括号左侧的子串
stringstream mid(s.substr(i+1,j-i-1));//中间部分的子串,转化成字符流方便用逗号分割
string right=s.substr(j+1);//右括号右侧的子串
string word;
while(getline(mid,word,',')){//拼接新的表达式进行递归
dfs(left+word+right);
}
};
dfs(expression);
for(auto &s:vis){
res.push_back(s.first);//从map中读出所有的key
}
return res;
}
};