【LeetCode-1087】花括号展开
问题
我们用一个特殊的字符串 S 来表示一份单词列表,之所以能展开成为一个列表,是因为这个字符串 S 中存在一个叫做「选项」的概念:
单词中的每个字母可能只有一个选项或存在多个备选项。如果只有一个选项,那么该字母按原样表示。
如果存在多个选项,就会以花括号包裹来表示这些选项(使它们与其他字母分隔开),例如 "{a,b,c}" 表示 ["a", "b", "c"]。
示例
输入: "{a,b}c{d,e}f"
输出: ["acdf","acef","bcdf","bcef"]
解答
class Solution {
public:
vector<string> expand(string s) {
for (int i = 0; i < s.size(); i++) { // 处理字符串使其便于dfs
if (s[i] == ',') continue;
string cur;
if (s[i] == '{') {
while (s[++i] != '}')
if (s[i] != ',') cur.push_back(s[i]);
sort(cur.begin(), cur.end());
}
else cur += s[i];
chs.push_back(cur);
}
dfs(0);
return res;
}
private:
vector<string> chs, res;
string ans;
void dfs(int pos) {
if (pos == chs.size()) {
res.push_back(ans);
return;
}
for (char ch : chs[pos]) {
ans.push_back(ch);
dfs(pos + 1);
ans.pop_back();
}
}
};
重点思路
举个例子,将{a, b} c {d, e} f
展开为{a, b} {c} {d, e} {f}
,然后就可以DFS了。