[LeetCode] Remove Invalid Parentheses 移除非法括号

Posted 2021-01-12 AcceptedLin

      [LeetCode] Remove Invalid Parentheses 移除非法括号

删除最小数量的无效括号，使得输入的字符串有效，返回所有可能的结果。

说明: 输入可能包含了除 ( 和 ) 以外的字符。

示例 1:

输入: "()())()"
输出: ["()()()", "(())()"]

示例 2:

输入: "(a)())()"
输出: ["(a)()()", "(a())()"]

示例 3:

输入: ")("
输出: [""]

DFS解法首先统计了多余的半括号的数量，用cnt1表示多余的左括号，cnt2表示多余的右括号，因为给定字符串左右括号要么一样多，要么左括号多，要么右括号多，也可能左右括号都多，比如")("。所以cnt1和cnt2要么都为0，要么都大于0，要么一个为0，另一个大于0。好，下面进入我们的递归函数，首先判断，如果当cnt1和cnt2都为0时，说明此时左右括号个数相等了，我们调用isValid子函数来判断是否正确，正确的话加入结果res中并返回即可。否则从start开始遍历，这里的变量start表示当前递归开始的位置，我们不需要每次都从头开始，会有大量重复计算。而且对于多个相同的半括号在一起，我们只删除第一个，比如"())"，这里有两个右括号，我们不管删第一个还是删第二个右括号都会得到"()"，没有区别，所以只用算一次就行了，我们通过和上一个字符比较，如果不相同，说明是第一个右括号，如果相同则直接跳过。此时来看如果cnt1大于0，说明此时左括号多，而如果当前字符正好是左括号的时候，我们可以删掉当前左括号，继续调用递归，此时cnt1的值就应该减1，因为已经删掉了一个左括号。同理，如果cnt2大于0，说明此时右括号多，而如果当前字符正好是右括号的时候，我们可以删掉当前右括号，继续调用递归，此时cnt2的值就应该减1，因为已经删掉了一个右括号。参见代码如下：

class Solution(object):
    def removeInvalidParentheses(self, s):
        """
        :type s: str
        :rtype: List[str]
        """
        if len(s) == 0:
            return [""]
        count1 = 0
        count2 = 0
        for str_num in s:
            count1 += (str_num=="(")
            if count1 == 0:
                count2 += (str_num==")")
            else:
                count1 -= (str_num==")")        
        start = 0
        res = []
        self.DFS(s, start, count1, count2, res)
        return res
    def DFS(self, s, start, count1, count2, res):
        if count1 == 0 and count2 == 0: #一样多
            if self.isValid(s):
                res.append(s)
            return
        else:
            for i in range(start, len(s)):
                if i != start and s[i] == s[i-1]:
                    continue #重复的只计算一次
                elif count1 > 0 and s[i] == "(": #左多
                    self.DFS(s[:i]+s[i+1:], i, count1-1, count2, res)
                elif count2 > 0 and s[i] == ")": #右多
                    self.DFS(s[:i]+s[i+1:], i, count1, count2-1, res)
    def isValid(self, string):
        count = 0
        for i in range(len(string)):
            if string[i] == "(":
                count += 1
            elif string[i] == ")":
                count -= 1
                if count < 0:
                    return False
        return count == 0

 

阅读更多