题目地址:
https://www.lintcode.com/problem/780/description
给定一个长 n n n的含有小括号的字符串 s s s(不含其它种类的括号),问要使得其括号是合法括号序列(在忽略非括号字符的情况下),至少要删多少个括号,返回所有删去多余括号所得字符串的合法方案。
先求出至少要删去多少个左括号和右括号,我们分别用 l l l和 r r r两个变量来存。合法括号等价于两个条件同时满足,即左右括号数相等并且对任意前缀左括号数都不小于右括号数。可以想象有一个栈,如果栈空的时候遇到右括号,则说明该右括号是要被删的,于是 r r r加上 1 1 1;如果最后栈不空,那么栈里剩余的左括号都是要被删的,于是 l l l就是栈的元素个数。实际求 l l l和 r r r的时候不需要真的维护一个栈,只需要对 s s s扫描一遍即可。不难证明这样计算出的 l l l和 r r r就是最少要删除的左括号和右括号数目。
接下来就是DFS做枚举,枚举所有删去 l l l个左括号和删去 r r r个右括号的情况,最后如果得到了一个括号合法的字符串就可以将答案加进去。这里可以做一些剪枝,比如说,对于连续的若干个左括号,删去其中任意 k k k个括号效果都是一样的,我们可以直接枚举删去括号的个数,而不是枚举删去的是哪些位置的括号;此外,如果当前得到的字符串里左括号数目少于右括号数目,那么右括号必须得删足够个,不枚举删的不够的情形。代码如下:
import java.util.ArrayList;
import java.util.List;
public class Solution {
/**
* @param s: The input string
* @return: Return all possible results
*/
public List<String> removeInvalidParentheses(String s) {
// Write your code here
List<String> res = new ArrayList<>();
// l和r分别是要删多少个左括号和右括号
int l = 0, r = 0;
for (int i = 0; i < s.length(); i++) {
if (s.charAt(i) == '(') {
l++;
} else if (s.charAt(i) == ')') {
if (l == 0) {
r++;
} else {
l--;
}
}
}
// cnt是左括号数减去右括号数
dfs(0, s, new StringBuilder(), 0, l, r, res);
return res;
}
private void dfs(int pos, String s, StringBuilder sb, int cnt, int l, int r, List<String> res) {
if (pos == s.length()) {
if (cnt == 0) {
res.add(sb.toString());
}
return;
}
char ch = s.charAt(pos);
if (ch != '(' && ch != ')') {
dfs(pos + 1, s, sb.append(ch), cnt, l, r, res);
sb.setLength(sb.length() - 1);
} else if (ch == '(') {
// 先求一下有多少个连续的左括号
int i = pos;
while (i < s.length() && s.charAt(i) == '(') {
i++;
}
// 先枚举将这些左括号全删去的情形,然后再枚举留1个,2个的情形,以此类推
l -= i - pos;
for (int j = i - pos; j >= 0; j--) {
if (l >= 0) {
dfs(i, s, sb, cnt, l, r, res);
}
sb.append('(');
cnt++;
l++;
}
// 上面的for循环结束后一共加了i - pos + 1个左括号,所以这里一次性恢复现场
sb.setLength(sb.length() - (i - pos + 1));
} else {
int i = pos;
while (i < s.length() && s.charAt(i) == ')') {
i++;
}
// 和上面相同,先枚举将这些左括号全删去的情形,然后再枚举留1个,2个的情形,以此类推
r -= i - pos;
for (int j = i - pos; j >= 0; j--) {
// 如果cnt < 0意味着左括号比右括号少,这时候不可能得到合法解的,直接略过
if (cnt >= 0 && r >= 0) {
dfs(i, s, sb, cnt, l, r, res);
}
sb.append(')');
cnt--;
r++;
}
sb.setLength(sb.length() - (i - pos + 1));
}
}
}
时间复杂度 O ( 2 n n ) O(2^nn) O(2nn),空间 O ( n ) O(n) O(n)。