HDU 5113 Black And White dfs+剪枝

题意：给你一个n*m的矩阵，现有K种颜料，要涂满整个矩阵。每种颜料可以涂的矩阵个数为c[i]，涂矩阵时要求相邻矩阵颜色不同。现求一种合法的涂法。

方法：dfs+剪枝

剪枝技巧，对未填方块进行黑白染色，则需当前颜料的个数大于(left+1)/2个，才能使满足要求。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<time.h>
#include<queue>
using namespace std;
const int N=6;
bool flag;
int num[N*N],n,m,K,left;
int col

;
void nextxy(int cx,int cy,int &nx,int &ny){
nx=cx;ny=cy;
ny++;
if(ny>m){
ny=1;
nx++;
}
}
bool dfs(int cx,int cy){
if(cx==n+1){
return true;
}
int ji=1;
for(int i=1;i<=K;i++)if(num[ji]<num[i])ji=i;
if(num[ji]>(left+1)/2)return  false;
int nx,ny;
nextxy(cx,cy,nx,ny);
for(int i=1;i<=K;i++)if(num[i]){
if(cx>1&&col[cx-1][cy]==i)continue;
if(cy>1&&col[cx][cy-1]==i)continue;
num[i]--;
col[cx][cy]=i;
left--;
if(dfs(nx,ny))return true;
left++;
num[i]++;
}
return false;
}
void work(){
int T,tt=1;
scanf("%d",&T);
while(T--){
printf("Case #%d:\n",tt++);
scanf("%d%d%d",&n,&m,&K);
for(int i=1;i<=K;i++)
scanf("%d",&num[i]);
left=n*m;
flag=dfs(1,1);
if(!flag){
puts("NO");
continue;
}
puts("YES");
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
printf("%d%c",col[i][j],j==m?'\n':' ');
}
}
}
}

int main(){
//freopen("data_in.txt","r",stdin);
work();
return 0;
}