HDU 5113 Black And White(搜索剪枝)

题意：给出n*m的格子，k种颜色及数量，现在向格子中填颜色，要求相邻的格子颜色不能相同，输出答案

一层层向格子中填颜色，加上一个小剪枝：某种颜色的数量大于剩下的格子数的一半则回溯，因为此时无论怎么填都会有颜色相邻

#include<cstring>
#include<string>
#include<iostream>
#include<queue>
#include<cstdio>
#include<algorithm>
#include<map>
#include<cstdlib>
#include<cmath>
#include<vector>
//#pragma comment(linker, "/STACK:1024000000,1024000000");

using namespace std;

#define INF 0x3f3f3f3f

int n,m,k;
int g[10][10];
int color[50];
int ok;
int dir[][5]= {{0,1},{0,-1},{1,0},{-1,0}};

bool check(int x,int y)
{
    for(int i=0; i<4; i++)
    {
        if(g[x][y]==g[x+dir[i][0]][y+dir[i][1]]) return false;
    }
    if(n*m-m*(x-1)-y==1) return true;
    for(int i=0; i<k; i++)
    {
        if(((n*m-m*(x-1)-y)+1)/2<color[i]) return false;
    }
    return true;
}


void dfs(int x,int y)
{
    if(x==n+1)
    {
        ok=1;
        return ;
    }
    if(y<=m)
    {
        for(int i=0; i<k; i++)
        {
            if(color[i])
            {
                g[x][y]=i;
                color[i]--;
                if(check(x,y))
                {
                    dfs(x,y+1);
                    if(ok) return ;
                }
                g[x][y]=-1;
                color[i]++;
            }
        }
    }
    else dfs(x+1,1);
}

int main()
{
    int ca=1;
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&n,&m,&k);
        for(int i=0; i<k; i++)
        {
            scanf("%d",&color[i]);
        }
        memset(g,-1,sizeof g);
        ok=0;
        dfs(1,1);
        printf("Case #%d:\n",ca++);
        if(!ok) printf("NO\n");
        else
        {
            printf("YES\n");
            for(int i=1; i<=n; i++)
            {
                for(int j=1; j<=m; j++)
                {
                    printf("%d",g[i][j]+1);
                    if(j!=m) printf(" ");
                }
                puts("");
            }
        }
    }
    return 0;
}