JZOJ5804. 【2018.08.12提高A组模拟】简单的序列

Description

从前有个括号序列 s，满足 |s| = m。你需要统计括号序列对 (p, q) 的数量。
其中 (p, q) 满足 |p| + |s| + |q| = n，且 p + s + q 是一个合法的括号序列。

Input

从文件 bracket.in 中读入数据。第一行两个正整数 n, m。
第二行一个长度为 m 的括号序列，表示 s。

Output

输出到文件 bracket.out 中。
输出一行一个整数，表示符合条件的 (p, q) 的数量对 10^9 + 7 取模的值。

Sample Input

【样例 1 输入】
4 1 (
【样例 2 输入】
4 4 (())
【样例 3 输入
4 3 (((

Sample Output

【样例 1 输出】
4
【样例 2 输出】
1
【样例 3 输出】
0

Data Constraint

对于 10% 的数据，n ≤ 20；
对于 25% 的数据，n ≤ 200；
对于另外 5% 的数据，n = m；
对于 55% 的数据，n − m ≤ 200；
对于 100% 的数据，1 ≤ m ≤ n ≤ 10^5, n − m ≤ 2000。

题解

关于括号序，很显然可以想到将（变为+1，）变为-1，然后维护前缀和。
显然，合法的括号序一定保证任何时候前缀和都是非负的。
设$f_{i,j}$表示当前做到括号序的第i位，前缀和为j的方案数。
转移很简单。

现在考虑加入s的方案数，
枚举p有多少位，前缀和，然后可以算出对应的q的位数还有前缀和。
因为p跟q其实本质相同，所以方案数都可以用f数组求出。
注意一点，要保证s的前缀和必须非负。

code

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#define ll long long
#define N 2003
#define M 103
#define P putchar
#define G getchar
using namespace std;
char ch;
void read(int &n)
{
    n=0;
    ch=G();
    while((ch<'0' || ch>'9') && ch!='-')ch=G();
    ll w=1;
    if(ch=='-')w=-1,ch=G();
    while('0'<=ch && ch<='9')n=(n<<3)+(n<<1)+ch-'0',ch=G();
    n*=w;
}

int max(int a,int b){return a>b?a:b;}
int min(int a,int b){return a<b?a:b;}
void write(ll x){if(x>9) write(x/10);P(x%10+'0');}

const int mo=1000000007;
int n,m,q,s,f[N][N],ans,mi;

int main()
{
    //freopen("bracket.in","r",stdin);
    //freopen("bracket.out","w",stdout);

    read(n);read(m);q=n-m;
    for(ch=G();ch!='(' && ch!=')';ch=G());
    for(s=0;ch=='(' || ch==')';ch=G())
        s=s+(ch=='('?1:-1),mi=min(s,mi);

    f[0][0]=1;
    for(int i=1;i<=q;i++)
        for(int j=0;j<=i;j++)
        {
            f[i][j]=f[i-1][j+1];
            if(j)f[i][j]=(f[i][j]+f[i-1][j-1])%mo;
        }

    for(int i=0;i<=q;i++)
    {
        for(int j=-mi;j<=q-s;j++)
            ans=(ans+(ll)f[i][j]*f[q-i][j+s]%mo)%mo;
    }

    printf("%d\n",ans);

    return 0;
}