动态规划之括号匹配(二)包含多种括号

题目描述

给你一个字符串，里面只包含"(",")","[","]"四种符号，请问你需要至少添加多少个括号才能使这些括号匹配起来。
如：
[]是匹配的
([])[]是匹配的
((]是不匹配的
([)]是不匹配的

动态规划思路
状态定义：使用二维dp，dp[i][j]表示从s[i]到s[j]的最少添加括号个数
初始化：单独一个字符时，dp[i][j]＝１，因为一个字符必定需要添加一个括号，其他值先赋值为0
状态转移方程：循环遍历区间长度(区间长度m从1到n)，每次考虑[i，j]内的区间(i从0开始,j=i+m)，

• 如果s[i]和s[j]匹配，那么dp[i][j]=dp[i+1][j-1]，也即括号里面部分需要添加的最少括号数．
• 分割情况：如果对区间[i,j]内的区间进行遍历分割，如果有更小的值，就采用更小的，也即
  dp[i][j]=min(dp[i][j], dp[i][k] + dp[k+1][j])

代码如下

def match(a, b):
    if a == '[' and b == ']':
        return True
    if a == '(' and b == ')':
        return True
    return False


def kuohao(s):
    n = len(s)
    dp = [[0] * n for _ in range(n)]
    for i in range(n):
        dp[i][i] = 1
    for m in range(1, n + 1):
        for i in range(n - m):
            j = i + m
            dp[i][j] = float('inf')
            if match(s[i], s[j]):
                dp[i][j] = dp[i + 1][j - 1]
            for k in range(i, j):
                dp[i][j] = min(dp[i][j], dp[i][k] + dp[k + 1][j])
    return dp[0][n - 1]


s = '((]]()'
print(kuohao(s))