[HDU 5113] Black And White (dfs+剪枝)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5113

题目大意：给你N*M的棋盘，K种颜色，每种颜色有c[i]个(sigma(c[i]) = N*M)，现在给棋盘染色，使得相邻的两个棋盘染成不同的颜色，并且把所有颜色用完。

因为棋盘最大为5*5的，因此可以考虑搜索+剪枝。

从左到右，从上到下看当前格子能够染成什么颜色。

有一个限制性条件，就是说如果当前棋盘的格子数量的一半小于一种颜色的数量时，那么就一定有两个相邻的棋盘被染成了相同的颜色。

因为假设棋盘容量为cap，那么我们间隔着放颜色，最多能够放cap/2个。

因此就可以用限制性条件来剪枝。

 #include <cstdio>
 #include <algorithm>
 #include <cstring>
 #include <vector>
 using namespace std;
 typedef long long LL;
 typedef pair<int,int> PII;
 typedef vector<int> vec;
 typedef vector<vec> mat;
 #define AA first
 #define BB second

 int T,N,M,K;
 int m[][];
 int c[];
 bool hasAns;

 bool C(int x,int y,int c){
     int row = true, column = true;
     if( x->=&&m[x-][y]==c ) column = false;
     if( y->= && m[x][y-]==c ) row = false;
     return row&&column;
 }

 void dfs(int x,int y,int cur){
     for(int i=;i<=K;i++){
         if( c[i]>(cur+)/ ) return;
     }
     if(hasAns) return;
     if( x==N+ ){
         hasAns = true;
         puts("YES");
         for(int i=;i<=N;i++) {
             for(int j=;j<=M;j++ ){
                 printf(j==M?"%d\n":"%d ",m[i][j]);
             }
         }
         return;
     }
     for(int i=;i<=K;i++)if( c[i] ){
         if( C(x,y,i) ){
             c[i]--;
             m[x][y] = i;
             if( y==M ) dfs(x+,,cur-);
             else dfs(x,y+,cur-);
             c[i]++;
             m[x][y] = ;
         }
     }
 }

 int main(){
     scanf("%d",&T);
     for(int cases = ; cases <= T; cases++){
         scanf("%d%d%d",&N,&M,&K);
         for(int i=;i<=K;i++){
             scanf("%d",&c[i]);
         }
         hasAns = false;
         printf("Case #%d:\n",cases);
         dfs(,,N*M);
         if(!hasAns) puts("NO");
     }
     return ;
 }