d[i][j]为字符串str[i~j]需要添加多少个括号,可以分为以下两种情况,当字符串形如’(s)’=和‘[s]’时就只要计算字符串s需要添加多少括号使之匹配,不必要管外面的已经匹配好的’()’和’[]’。当字符至少有两个时可以找个分界点K计算d(i,k)+d(k+1,j)。但不管字符是否满足第一条,都要尝试第二种,例如字符’[][]’,如果不尝试第二种转移,就会转移到’][‘,然后就只能添加两个括号了。
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <string>
#include <string.h>
using namespace std;
const int maxn = 150;
int d[maxn][maxn];
char str[maxn];
int len;
bool match(char a, char b)
{
if ((a == '('&&b == ')')||(a=='['&&b==']'))
return true;
return false;
}
void dp()
{
for (int i = 0; i < len; i++)
d[i][i] = 1;
for (int i = len - 2; i >= 0; i--)
{
for (int j = i + 1; j < len; j++)
{
d[i][j] = (1 << 30);
if (match(str[i], str[j]))
{
d[i][j] = min(d[i][j], d[i + 1][j - 1]);
}
for (int k = i; k < j; k++)
{
d[i][j] = min(d[i][j], d[i][k] + d[k + 1][j]);
}
}
}
}
void print(int i, int j)
{
if (i > j)
return;
if (i == j)
{
if (str[i] == '(' || str[j] == ')')
printf("()");
else
printf("[]");
return;
}
int ans = d[i][j];
if (match(str[i],str[j])&&ans == d[i + 1][j - 1])
{
printf("%c", str[i]);
print(i + 1, j-1);
printf("%c", str[j]);
return;
}
for (int k = i; k < j; k++)
{
if (ans == d[i][k] + d[k + 1][j])
{
print(i, k);
print(k + 1, j);
return;
}
}
}
int main()
{
gets(str);
memset(d, -1, sizeof(d));
len = strlen(str);
dp();
print(0, len - 1);
cout << endl;
return 0;
}