AtCoder Grand Contest 017 F - Zigzag 状压dp

题意

给出n和m，求有多少个m位二进制数，满足对于任意相邻的两个，都有i+1的前j位的和不小于i的前j位的和。此外有若干个二进制位上的数已经被确定。
n,m<=20

分析

很容易想到可以设f[i,s]表示第i个数为s时的方案，转移则需要枚举下一位是什么。
这样子显然会超时，考虑如何优化。
既然不能枚举下一个，那就考虑这一个数和上一个数一起改变好了。
设f[i,j,s]表示做到第i个二进制数，前j位和s的前j位相等，且在以后一直不会跑到s的左边的方案。
如果第j+1位放的数和s相同，就不用改变s。
若s为1，下一位放0显然不合法。
若s为0，下一位放1的话，考虑找到最小的一个k满足$k>j$且s的第k位为1，然后把s的第j+1位变为1，第k位变为0，然后就可以转移过去了。
如果s的后面几位全为0，则不改变就好。
这样就可以过了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=25;
const int MOD=1000000007;

int n,m,p,c[N][N],f[2][21][1048580],bin[N];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void updata(int &x,int y)
{
    x+=y;x-=x>=MOD?MOD:0;
}

int modify(int s,int i)
{
    if (s&bin[i]) return s;
    int w=s&(bin[n]-bin[i]);
    if (w) s-=w&(-w);
    return s+bin[i];
}

int main()
{
    n=read()-1;m=read();p=read();
    for (int i=1;i<=m;i++)
        for (int j=1;j<=n;j++)
            c[i][j]=-1;
    for (int i=1;i<=p;i++)
    {
        int x=read(),y=read(),z=read();
        c[x][y]=z;
    }
    bin[0]=1;
    for (int i=1;i<=n;i++) bin[i]=bin[i-1]*2;
    int now=0;
    f[0][0][0]=1;
    for (int i=1;i<=m;i++)
    {
        for (int j=0;j<n;j++)
            for (int s=0;s<bin[n];s++)
            {
                if (!f[now][j][s]) continue;
                if (c[i][j+1]<=0&&!(s&bin[j])) updata(f[now][j+1][s],f[now][j][s]);
                if (c[i][j+1]!=0) updata(f[now][j+1][modify(s,j)],f[now][j][s]);
            }
        memset(f[now^1],0,sizeof(f[now^1]));
        for (int s=0;s<bin[n];s++) updata(f[now^1][0][s],f[now][n][s]);
        now^=1;
    }
    int ans=0;
    for (int s=0;s<bin[n];s++) updata(ans,f[now][0][s]);
    printf("%d",ans);
    return 0;
}