CF293B Distinct Paths 搜索

考试 T2,没想到真的就是一个大爆搜啊~ 

搜索时几个剪枝:

1.剩余可用颜色要大于等于剩余步数 

2.颜色从没用过的里找,然后用二进制状压存储. 

3. 最重要的一个剪枝:如果一个颜色在之前序列中未出现过,且在之后的位置也不会强制出现,则所有这种情况的贡献都是相同的,只需算第一次的贡献,然后后面的依次累加即可.   

code: 

#include <bits/stdc++.h>          
using namespace std;  
void setIO(string s) 
{
    string in=s+".in";  
    string out=s+".out";  
    freopen(in.c_str(),"r",stdin);            
    // freopen(out.c_str(),"w",stdout);   
}
const int mod=1e9+7;   
int lg[1040],n,m,k,a[23][23],f[23][23],v[23];      
int dfs(int x,int y) 
{
    if(y==m+1) ++x,y=1;                                     
    if(x==n+1) return 1;   
    int s=f[x-1][y]|f[x][y-1],calc=-1,ret=0;  
    int S=~s&((1<<k)-1);           
    if(n+m-x-y+1>lg[S]) return 0;     
    for(int tmp=0;tmp<k;++tmp) 
    {
        if(S&(1<<tmp)) 
        {
            if(a[x][y]==0||a[x][y]==tmp+1) 
            {
                v[tmp+1]++;  
                f[x][y]=s|(1<<tmp);     
                if(v[tmp+1]==1) 
                {   
                    if(calc==-1) calc=dfs(x,y+1);   
                    ret+=calc;   
                } 
                else ret+=dfs(x,y+1);  
                if(ret>=mod) ret-=mod;      
                v[tmp+1]--;                                                                   
            }           
        }
    }
    return ret;     
}
int main() 
{
    // setIO("input");    
    int i,j;  
    for(i=1;i<1024;++i) lg[i]=lg[i>>1]+(i&1);    
    scanf("%d%d%d",&n,&m,&k);   
    if(n+m-1>k) { printf("0\n");  return 0; }  
    for(i=1;i<=n;++i) 
    {
        for(j=1;j<=m;++j) 
            scanf("%d",&a[i][j]), v[a[i][j]]++;     
    }
    printf("%d\n",dfs(1,1));    
    return 0;  
}

  

posted @ 2019-10-29 14:31  EM-LGH  阅读(168)  评论(0编辑  收藏  举报