【Lintcode】780. Remove Invalid Parentheses

题目地址：

https://www.lintcode.com/problem/780/description

给定一个长$n$的含有小括号的字符串$s$（不含其它种类的括号），问要使得其括号是合法括号序列（在忽略非括号字符的情况下），至少要删多少个括号，返回所有删去多余括号所得字符串的合法方案。

先求出至少要删去多少个左括号和右括号，我们分别用$l$和$r$两个变量来存。合法括号等价于两个条件同时满足，即左右括号数相等并且对任意前缀左括号数都不小于右括号数。可以想象有一个栈，如果栈空的时候遇到右括号，则说明该右括号是要被删的，于是$r$加上$1$；如果最后栈不空，那么栈里剩余的左括号都是要被删的，于是$l$就是栈的元素个数。实际求$l$和$r$的时候不需要真的维护一个栈，只需要对$s$扫描一遍即可。不难证明这样计算出的$l$和$r$就是最少要删除的左括号和右括号数目。

接下来就是DFS做枚举，枚举所有删去$l$个左括号和删去$r$个右括号的情况，最后如果得到了一个括号合法的字符串就可以将答案加进去。这里可以做一些剪枝，比如说，对于连续的若干个左括号，删去其中任意$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(n)$。