HDU 5113 Black And White (搜索剪枝)

题意：

给一个N*M的格子，给出K种颜色以及每种颜色的数目，保证数目和相加 = N*M，求一种方案使得上下左右相邻的格子颜色都不相同。

思路：

搜索

暴力剪枝：

剩下的格子的数量+1必需是剩余最多种类棋子的两倍

代码：

#include <bits/stdc++.h>
using namespace std;
int a[30];
int vis[6][6];
int dirx[]={0,0,1,-1};
int diry[]={1,-1,0,0};
int n,m,k;
bool judge(int left){
    for(int i=1;i<=k;i++) if((left+1)<a[i]*2) return 0;
    return 1;
}
bool check(int x,int y,int col){
    for(int i=0;i<4;i++){
        int nowx=x+dirx[i],nowy=y+diry[i];
        if(nowx>=n||nowy>=m||nowx<0||nowy<0) continue;
        if(col==vis[nowx][nowy]) return 0;
    }
    return 1;
}
bool dfs(int x,int y,int col,int left){
    vis[x][y]=col;
    a[col]--;
    left--;
    if(left==0) return 1;
    if(judge(left))
        for(int i=0;i<4;i++){
            int nowx=x+dirx[i],nowy=y+diry[i];
            if(nowx>=n||nowy>=m||nowx<0||nowy<0||vis[nowx][nowy]) continue;
            for(int j=1;j<=k;j++) if(a[j]&&check(nowx,nowy,j)) if(dfs(nowx,nowy,j,left)) return 1;
        }
    vis[x][y]=0;
    a[col]++;
    return 0;
}
int main(){
    int T;
    scanf("%d",&T);
    for(int Case=1;Case<=T;Case++){
        scanf("%d%d%d",&n,&m,&k);
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=k;i++) scanf("%d",&a[i]);
        for(int i=1;i<=k;i++){
            if(dfs(0,0,i,n*m)){
                break;
            }
        }
        printf("Case #%d:\n",Case);
        if(vis[0][0]){
            puts("YES");
            for(int i=0;i<n;i++){
                for(int j=0;j<m;j++)
                    if(j!=m-1)
                        printf("%d ",vis[i][j]);
                    else
                        printf("%d\n",vis[i][j]);
            }
        }else{
            puts("NO");
        }
    }
}