[JLOI2015]bzoj 4005 骗我呢 - 组合数学 - dp

莫名其妙T了……
考虑朴素dp，设dp[i,j]表示前i行j这个数字没出现，那么显然下一行只能是[j-1,m]的数字没出现，也就是$dp[i,j]=\sum_{k=0}^{min(j+1,m)}dp[i-1,k]$，如果$0<j<m$那么$dp[i,j]=dp[i,j-1]+dp[i-1,j+1]$，然后那个转移的图画出来，然后人为补充上-1这个点并平移每一行，最后发现就是在这种网格图上向右n+m+1向下n次的方案数。将向右看作+1向下看作-1那么任意时刻前缀和>=0并且<=m+1=(n+m+1)-(n)。
问题转化为求F(n,m)={n个+1和m个-1任意时刻前缀和>=0并且<=n-m的方案数}。
考虑容斥，用总数减去先超过上界（记做<0），再减去先超过下界（记做>n-m），以减去先超过下界为例，可以先减去最后超过下界(<0)的方案（如多次则以最后一次为准），再加上最后是先>n-m再<0的，再减去最后是先< 0然后>n-m然后< 0……
考虑没一个怎么算，具体方法和卡特兰数有点类似：这个过程把序列划分成若干段，从后向前翻转即可。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<assert.h>
#define mod 1000000007
#define lint long long
#define N 20000010
#define A(x,y) (swap(x,y),x--,y++)
#define B(x,y) (swap(x,y),x+=n-m+1,y-=n-m+1)
#define P(x) (x<0?x+=mod:0)
#define Q(x) (x>=mod?x-=mod:0)
#define debug(x) cerr<<#x<<"="<<x
#define sp <<" "
#define ln <<endl
using namespace std;
int f[N],fi[N];
inline int fast_pow(int x,int k,int ans=1)
{   for(;k;k>>=1,x=(lint)x*x%mod) (k&1)?ans=(lint)ans*x%mod:0;return ans;   }
inline int prelude(int n)
{
    for(int i=f[0]=1;i<=n;i++) f[i]=(lint)f[i-1]*i%mod;
    fi[n]=fast_pow(f[n],mod-2);
    for(int i=n-1;i>=0;i--) fi[i]=fi[i+1]*(i+1ll)%mod;
    return 0;
}
inline int C(int n,int m)
{
    if(n<0||m<0||n<m) return 0;
    return (lint)f[n]*fi[m]%mod*fi[n-m]%mod;
}
inline int F(int n,int m,int ans=0)//n +1, m -1;
{
    for(int x=n,y=m;x>=0&&y>=0;)
        A(x,y),ans-=C(x+y,x),P(ans),
        B(x,y),ans+=C(x+y,x),Q(ans);
    for(int x=n,y=m;x>=0&&y>=0;)
        B(x,y),ans-=C(x+y,x),P(ans),
        A(x,y),ans+=C(x+y,x),Q(ans);
    return ans+=C(n+m,n),Q(ans),ans;
}
int main()
{
    prelude(N-1);int n,m;scanf("%d%d",&n,&m);
    return !printf("%d\n",F(n+m+1,n));
}