jzoj4012 [CF293B]Distinct Paths 搜索+剪枝

Description

---

小首有一个n*m的木板，一些块已经被涂上给出的k种颜色中的一种。你需要把每个没涂色的块涂色使得从左上角到右下角的每条路径都不会经过两个颜色一样的块。路径只能向右或向下走。
输出答案%1000000007。

对于40%的数据，1<=n,m<=3
对于100%的数据，1<=n,m<=1000,1<=k<=10

Solution

---

重启了一波然后重写了一波，差评

早上一直在刚T1和T2，结果忽略了这个送命水题GG
显然n+m-1<=k才有解，然后就能爆搜了。k很小，随便用二进制记录一下状态即可。还可以用lowbit加速枚举
考虑剪枝
1. 可行性剪枝。若剩下步数大于可用颜色就剪掉
2. 对称性剪枝。如果当前是第一次填颜色x，那么把x换成其他第一次填的颜色结果不变

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (register int i=st;i<=ed;++i)
#define lowbit(x) ((x)&(-(x)))

const int MOD=1000000007;

int rc[7][7],n,m,k,ans;
int rec[7][7],used[11];

__attribute__((optimize("O2")))
inline int dfs(int x,int y) {
    if (y>m) y=1,x++;
    if (x==n+1) return 1;
    int now=rec[x-1][y]|rec[x][y-1],tmp=-1;
    int ret=0,tot=0;
    rep(i,1,k) tot+=!((1<<i-1)&now);
    if (n+m-x-y+1>tot) return 0;
    rep(i,1,k) {
        if ((1<<i-1)&now) continue;
        if (!rc[x][y]||rc[x][y]==i) {
            rec[x][y]=now|(1<<i-1);
            if (++used[i]==1) {
                if (tmp==-1) tmp=dfs(x,y+1);
                ret+=tmp;
            } else ret+=dfs(x,y+1);
            ret%=MOD;
            --used[i];
        }
    }
    return ret;
}

int main(void) {
    int cnt=0; scanf("%d%d%d",&n,&m,&k);
    if (n+m-1>k) {puts("0"); return 0;}
    rep(i,1,n) rep(j,1,m) {
        scanf("%d",&rc[i][j]);
        cnt+=(!rc[i][j]); used[rc[i][j]]++;
    }
    printf("%d\n", dfs(1,1));
    return 0;
}