典型的枚举型题目,常用解法就是回溯法加上剪枝优化。
回溯就是一个一个试,最后舍去不符合要求的,剪枝的目的是让我们能够尽早地判断出不符合要求的例子,尽早地舍弃。
括号匹配不符合要求的情况可以这么理解:每一个左括号右边必然有一个右括号跟它匹配,而且一个括号只能匹配一个括号。
可以把"("
视为+1
, ")"
视为-1
, 那么我们从左往右计数,设计数值为key
,n 为括号的对数,就有:
- 字符串末尾的时候 key == 0,才是有效的字符串
- 任何时候 key 都不可以小于 0
- 任何时候 key 都不可以大于 n
ok,那初步的回溯思路以及简单的剪枝条件已经有了,虽然还是暴力了点:
class Solution {
public:
vector<string> res;
void backtrack(int n, int key, string cur){
if(key < 0 || key > n) return;
if(cur.size() == 2*n){
if(key == 0) res.push_back(cur);
return;
}
backtrack(n, key-1, cur+')');
backtrack(n, key+1, cur+'(');
}
vector<string> generateParenthesis(int n) {
if(n == 0) return vector<string>{""};
string cur("(");
backtrack(n, 1, cur);
return res;
}
};
下面我们接着优化,为了方便,我们不在仅仅使用一个变量 key 来记录左右括号的总和,而是使用两个变量left
, right
来分别记录左右括号的数量,那么有两条剪枝规则:
- 当 left== n 的时候,就不能再放左括号
- 只有当 right < left 的时候,才能接着放右括号
class Solution {
public:
vector<string> res;
void backtrack(int n, int left, int right, string cur){
if(cur.size() == 2*n){
res.push_back(cur);
return;
}
if(left < n) backtrack(n, left+1, right, cur+'(');
if(right < left) backtrack(n, left, right+1, cur+')');
}
vector<string> generateParenthesis(int n) {
if(n == 0) return vector<string>{""};
string cur("(");
backtrack(n, 1, 0, cur);
return res;
}
};