【WC2016模拟】计数系统(stones)

Description

n,k<=2000,T<=50

Solution

是不是很容易想到数位Dp？
讲一下我考场上想到的辣鸡做法，从中间向两边对称着做，满足第三个条件必然在某个时刻出现左端点为0，右端点为1，分成两段Dp。
特别难写而且自带一个大常数在50组数据的加强下可以达到接近N^3
膜拜一下题解：
先考虑字典序严格小于，枚举一个X把它变成I，然后考虑这一段前缀对答案的贡献。
既然要反串大于它，再枚举这段前缀其中的一个I变成X作为后缀，中间一段的答案可以用组合数预处理。
相等的情况也可以用组合数预处理。
注意如果前缀长度大于串长的一半要特判。

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;

typedef long long ll;

const int N=2*1e3+5,Mo=1e9+7,Inv2=5*1e8+4;

int C[N+5][N+5],F[N+5][N+5],G[N+5][N+5];
int n,k,Ans,s[N+5],a[N+5];

void inc(int &x,int y) {
    x=x+y;
    if (x>=Mo) x=x-Mo;
}

void prepare() {
    fo(i,0,N) {
        C[i][0]=1;
        fo(j,1,i) {
            C[i][j]=C[i-1][j];
            inc(C[i][j],C[i-1][j-1]);
        }
    }
    fo(i,0,N) 
        fo(j,0,N) {
            if (j&1) F[i][j]=C[i+1][j];
            if (j) inc(F[i][j],F[i][j-1]);
        }
    fo(i,0,N)
        fo(j,0,N) {
            if (j) G[i][j]=G[i][j-1];
            if (!(j&1)) {
                inc(G[i][j],C[i+1][j]);
                inc(G[i][j],C[(i+1)/2][j/2]);
            }
        }
    fo(i,0,N) fo(j,0,N) G[i][j]=(ll)G[i][j]*Inv2%Mo;
}

int calc(int st) {
    int res=0,cnt=0,tmp=0;

    fo(i,2,st) cnt+=s[i]^s[i-1];
    if (2*st<=n&&k>=2*cnt) res=G[n-2*st][k-2*cnt];

    if (2*st>n) {
        fd(i,n,st+1) a[i]=s[n-i+1];
        fo(i,1,st) a[i]=s[i];
        bool ok=1;
        fo(i,1,n/2) 
            if (a[i]>a[n-i+1]) {ok=0;break;}
            else if (a[i]<a[n-i+1]) break;
        int cur=0;
        fo(i,2,n) cur+=a[i]^a[i-1];
        if (cur<=k&&ok) res++;
    }

    fo(i,1,st) {
        if (st<n-i+1&&!s[i]) {
            int now=tmp+((i>1)&!s[i-1]);
            if (k>=now+cnt) inc(res,F[n-i-st][k-now-cnt]);
        } else if (st>=n-i+1) break;
        if (i>1) tmp+=s[i]^s[i-1];
    }

    return res;
}

void solve() {
    while (scanf("%d%d",&n,&k)!=EOF) {
        char ch;
        for(ch=getchar();ch!='X'&&ch!='I';ch=getchar());
        fo(i,1,n) {
            s[i]=ch=='X';
            ch=getchar();
        }
        Ans=0;
        fo(i,1,n) 
            if (s[i]) {
                s[i]=0;
                inc(Ans,calc(i));
                s[i]=1;
            }
        printf("%d\n",Ans+1);
    }
}

int main() {
    prepare();
    solve();
    return 0;
}