Atcoder Grand Contest 017 F Zigzag

Zigzag

Problem Statement

有一些点，共$n$行，第$i$行有$i$个点，记第$i$行第$j$个点为$(i,j)$，$(i,j)$每次可以往左下走到$(i+1,j)$，或往右下走到$(i+1,j+1)$，现在有$m$条由$(1,1)$走到第$n$行的路径，设$X_{p,i}$表示第$p$条路径经过了点($i$,$X_{p,i}$)，要求对于$1$<=$p$<$m$，$1$< $i$ <=$n$，必须要有$X_{p,i}$<=$X_{p+1,i}$，接下来给出若干条件形如($a$,$b$,$c$)表示强制$X_{a,b+1}$=$X_{a,b}$+$c$，求这$m$条路径在满足条件下的方案数模10^9+7。

Data Constraint

$n$,$m$≤$20$

Solution

我们显然可以把一条路径表示成一个$n$-$1$位的二进制数。
假设已经求出第$i$-$1$条的状态$S$的方案数$f_{i-1,S}$，再确认第$i$条路径的状态$S'$，设它们的前$j$-$1$位都相同，接着第$j$位开始就不同了，$S$的第$j$位为$0$，$S'$的为$1$,那么显然我们可以把$S$在第$j$位后的第一个$1$变成$0$,把第$j$位的$0$变为$1$,设得到的新状态为$T$,那么就能同层转移到$f_{i-1,T}$，显然这样变动$S$是正确的。
若在第$j$位后没有$1$了，那接下来可以乱走了都没问题了，根据这个就可$dp$了。

时间复杂度$O$($nm2^{n-1}$)，详细实现可以参考代码。

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

#define fo(i,j,l) for(int i=j;i<=l;++i)
#define fd(i,j,l) for(int i=j;i>=l;--i)

using namespace std;
typedef long long LL;
const LL N=21,M=(1<<19)+3,mo=1e9+7;

int dir[N][N],m2[N<<1],nex[N<<1];
int n,m,k,a,b,c;
LL f[2][N][M],g[N][M];

int main()
{
    cin>>n>>m>>k; --n;
    fo(i,1,k){
        scanf("%d%d%d",&a,&b,&c);
        dir[a][b]=c+1;
    }
    int u=0,v;
    f[0][0][0]=1;
    m2[1]=1;
    fo(i,2,n+1)m2[i]=m2[i-1]<<1;
    int zd=(m2[n]<<1)-1;
    fo(p,1,m){
        v=u^1;
        fo(l,0,n)fo(i,0,zd)f[v][l][i]=0;
        fo(l,0,n)fo(i,0,zd)g[l][i]=0;
        fd(i,zd,0){
            LL ok=f[u][0][i]%mo; nex[n]=n+1;
            fd(l,n,2)if(i&m2[l])nex[l-1]=l;else nex[l-1]=nex[l];
            if(!i)g[0][0]=ok;
            fo(l,1,n){
                if((i&m2[l])==0&&dir[p][l]!=1&&nex[l]!=n+1)
                f[u][l][(i^m2[nex[l]])^m2[l]]=f[u][l][(i^m2[nex[l]])^m2[l]]+ok;
                ok=(f[u][l][i]+ok)%mo;
                if(dir[p][l]&&(((i&m2[l])>0)-(dir[p][l]==2)))break;
                if(nex[l]==n+1&&(i&m2[l])>0)g[l][i]=ok;
            }
        }
        fo(i,0,n-1)
        fo(l,0,zd)if(g[i][l]){
            g[i][l]=g[i][l]%mo;
            if(dir[p][i+1]!=2)g[i+1][l]=g[i+1][l]+g[i][l];
            if(dir[p][i+1]!=1)g[i+1][l^m2[i+1]]=g[i+1][l^m2[i+1]]+g[i][l];
        }
        fo(i,0,zd)f[v][0][i]=g[n][i];
        u=v;
    }
    LL ans=0;
    fo(i,0,zd)ans=(ans+f[v][0][i])%mo;
    cout<<ans;
}