12届蓝桥杯省赛c++b组 J题 括号序列

这次要讲的前几个星期刚比完的蓝桥杯c++b组J题：括号序列。本次比赛我也参加了，但是这道题我是dfs求解的，所以都只是拿了少部分的分，我比赛时的代码就不展示了，因为时间复杂度很高，所以我就直接讲解正解应该怎么写了。

先上题目：

给定一个括号序列，要求尽可能少地添加若干括号使得括号序列变得合法，当添加完成后，会产生不同的添加结果，请问有多少种本质不同的添加结果。

两个结果是本质不同的是指存在某个位置一个结果是左括号，而另一个是右括号。

例如，对于括号序列 ((()，只需要添加两个括号就能让其合法，有以下几种不同的添加结果：()()()、()(())、(())()、(()()) 和 ((()))。

输入格式
输入一行包含一个字符串 s，表示给定的括号序列，序列中只有左括号和右括号。

输出格式
输出一个整数表示答案，答案可能很大，请输出答案除以 1000000007 (即 10^9+7) 的余数。

数据范围
对于 40% 的评测用例，|s|≤200。 对于所有评测用例，1≤|s|≤5000。

Sample Input：

((()

Sample output：

5

这道题呢就是一个括号匹配的问题，其实这种题目也是非常经典的一类题目。因为我们刚开始认识栈的时候，老师都会举括号匹配的例子来说明栈的应用。这种括号的题目是必须遵守一个规则的：在任何时候，一个合法的括号序列都满足左括号的数量大于等于右括号的数量！

举一个例子：()))这个括号序列在 i=2，3的时候都是不合法的（下标从0开始计数），而((()这个括号序列在当前都是合法的，这是因为左括号匹配右括号，如果只有右括号没有与之对应的左括号的话就必须得在这个右括号之前加上一个左括号。而如果左括号比右括号多的话，在这个位置之后可能还会有右括号和前面的左括号对应，所以我们不一定非得在这个时候加上右括号。

那么解决这个题目的算法是什么呢？这道题存在大量的重叠子问题，如果dfs一个个求的话，我们的时间复杂度会很高，再来看看我们的数据范围1≤|s|≤5000，也就是我们必须得设计一个O（n^2）以内的算法才能通过所有的测试用例，所以这道题可以使用动态规划来解决。

我们该如何定义状态？其实我们可以先看左括号，用f[i][j]来表示我们遍历到第 i 个括号，此时左括号比右括号多 j 个的时候的添加括号方案数。先算出添加左括号的方案数，再从右到左算出添加右括号的方案数，最后两个相乘就是我们的结果。

其实对于这道题来说，我们不需要写两个函数分别求左括号和右括号方案数，我们可以先算出左括号方案数，然后把这个字符串进行翻转，然后将左括号换成右括号，右括号换成左括号再求一次这个函数就是右括号的方案数。这是为什么呢？我们可以倒着看一下，上面的定理对于倒着依旧是相对成立的，倒着看的时候，一个合法的括号序列都满足右括号的数量大于等于左括号的数量。具体可以把上面的论点逆着推一下。

可能会有人有疑问，为什么两个方案数是相乘的关系，对于两个括号中间加入左右括号，难道里面的括号重新排列不是另一种方案吗？其实他们之间是相对独立的，我们可以这样想，…… |…… 如果我们在 | 的地方加入一个左括号，一个右括号的话，只能形成 ) ( 这样的序列，不能形成这样()，这是因为如果是后者的话，这算一对括号，可以抵消掉，还多了一对，题目的要求是尽可能少地添加若干括号，所以无论怎样，都只会有一种排列方式。

既然这样的话，我们怎么保证添加的时候不重复呢？也就是(((我们算成了好几种，其实只有一种。我们可以以右括号为界限，如果以右括号为界限，比如这样……）……）……，其实……都是左括号，他们无论怎样排列都是一样的。

至此我们的状态转移方程就好写了，我们只考虑在右括号的前面添加n个左括号，因为其他时候添加左括号其实都一样(里面全是左括号的话怎么排列都是一样的)。当我们遇到左括号的时候非常简单，直接把这个左括号加进来，就是比前一个状态多了1个左括号，f[i][j]=f[i-1][j-1]。比如我们前面左括号比右括号多5个，那么现在我们遇到的是左括号,我们的左括号就比右括号多了6个。

当我们遇到的是右括号的时候，我们考虑在右括号前面添加n个左括号，从0开始枚举，我们前面加0个左括号就是f[i][j]=f[i-1][j+1]，因为前面左括号比右括号多j+1个，现在遇到右括号了，我们不加左括号的情况下就多了j个。如果加1个左括号的话f[i][j]=f[i-1][j]，因为前面左括号比右括号多j个，现在遇到右括号了，我们加1个左括号的情况下就多了j个，以此类推。

所以f[i][j]就是前面添加左括号的所有情况相加，即f[i][j]=f[i-1][j+1]+f[i-1][j]+f[i-1][j-1]+……+f[i-1][0]。我们会发现，如果这样子加的话我们时间复杂度就是O(n)，会使得整体时间复杂度变成O(n^3)会超时。

不过我们可以发现f[i][j-1]=f[i-1][j]+f[i-1][j]+f[i-1][j-1]+……+f[i-1][0]，这样的话f[i][j]=f[i-1][j+1]+f[i][j-1]就会是O(1)的时间复杂度了，思路就到这里了，我附上代码：

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
using LL=long long;
const int N = 5005;
int f[N][N];
int mod=1e9+7;
string s;
int n;
LL get(){
    memset(f,0,sizeof f);
    f[0][0]=1;
    for(int i=1;i<=n;i++){
        if(s[i-1]=='('){
            for(int j=1;j<=n;j++)
                f[i][j]=f[i-1][j-1];
        }
        else{
            f[i][0]=(f[i-1][1]+f[i-1][0])%mod;
            for(int j=1;j<=n;j++)
                f[i][j]=(f[i-1][j+1]+f[i][j-1])%mod;
        }
    }
    for(int i=0;i<=n;i++)
        if(f[n][i])
            return f[n][i];
    return -1;
}
int main(){
    cin>>s;
    n=s.size();
    LL x=get();
    reverse(s.begin(),s.end());
    for(int i=0;i<n;i++){
        if(s[i]==')')
            s[i]='(';
        else
            s[i]=')';
    }
    LL y=get();
    cout<<(x*y)%mod;
}

base case就是f[0][0]=1，即我们看到前0个括号，左括号比右括号多0个的方案数为1，也是符合实际的。注意在遇到右括号的时候我们要特别处理一下f[i][0]，这是因为根据上面的状态转移方程，当j=0时f[i][j-1]会越界，我们就用上面O(n)的方法对f[i][0]进行特殊处理，因为只有两项，所以也是O(1)的时间复杂度。

值得一提的是，最后求答案的时候我们得找出f[n][i]中不为0的作为答案。有些同学就有疑问了，为什么不取f[n][0]？
我举一个例子：((((
i=1时，f[1][1]=1，其余f[1][x]都为0；
i=2时，f[2][2]=1，其余f[2][x]都为0；
i=3时，f[3][3]=1，其余f[3][x]都为0；
i=4时，f[4][4]=1，其余f[4][x]都为0；
所以如果我们求f[n][0]的时候方案数就是0，这是不对的。根本不存在左括号比右括号多0个情况，这是因为我们只添加左括号，不添加右括号。

不过不要忘记对结果取模哦。

补：

应粉丝要求，我把比赛时写的dfs代码附上，不过会TLE。各位参考一下就好

#include <iostream>
#include <cstring>
#include <algorithm>
#include <unordered_set>
using namespace std;

int n;
string s;
unordered_set<string> ret;

void dfs(int l,int r,int al,int ar,int idx,string cur){
    if(idx==n){
        if(!al&&!ar)
            ret.insert(cur);
        return;
    }
    if(al)
        dfs(l+1,r,al-1,ar,idx,cur+'(');
    if(ar&&l>r)
        dfs(l,r+1,al,ar-1,idx,cur+')');
    if(s[idx]=='(')
        dfs(l+1,r,al,ar,idx+1,cur+'(');
    else if(l>r)
        dfs(l,r+1,al,ar,idx+1,cur+')');
}

int main(){
    cin>>s;
    n=s.size();
    int al=0,ar=0,l=0;
    for(char c:s){
        if(c=='('){
            l++;
        }
        else if(c==')'){
            if(l==0)
                al++;
            else{
                l--;
            }
        }
    }
    ar=l;
    dfs(0,0,al,ar,0,"");
    cout<<ret.size();
}

这道题就讲解到这里啦，继续加油：）