hdu 5113 Black And White

题意：  有一个n*m的矩阵，我们现在有k种不同的颜色，问能够用这k种颜色把矩阵涂满，并且相邻的格子颜色不同，每种颜色的数量会给出。

分析：看数据范围可以看到（0<n,m<=5）,所以感觉可以直接深搜暴力+剪枝。

剪枝：对于一个矩阵来说，当一种颜色的数量>(n*m+1)/2的时候，就代表一定会有相邻颜色相同，否则一定是可以构造符合条件的矩阵的，所以我们在中dfs中，可以通过判断剩余颜色中是否有一种的数量是大于剩余格数的一半，来剪枝。

代码：

 

#include<bits/stdc++.h>
using namespace std;

int n,m,k;
int h[30];
int ans[10][10];
bool flag;

void dfs(int x,int y,int sum)
{
    if(flag) return ;
    if(x==n&&y==m+1)
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=m;j++)
            {
                printf("%d",ans[i][j]);
                if(j!=m) printf(" ");
            }
            printf("\n");
        }
        flag=true;
        return;
    }
    for(int i=0;i<k;i++)
    {
        if(h[i]>(n*m-sum+1)/2)
        {
            return;
        }
    }
    if(y==m+1) x++,y=1;
    for(int i=0;i<k;i++)
    {
        if(h[i]&&ans[x-1][y]!=i+1&&ans[x][y-1]!=i+1)
        {
            h[i]--;
            ans[x][y]=i+1;
            dfs(x,y+1,sum+1);
            h[i]++;
            ans[x][y]=0;
        }
    }
}

int main()
{
    int T ,u=1;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&n,&m,&k);
        memset(ans,0,sizeof(ans));
        int maxsum=-1;
        for(int i=0;i<k;i++)
            scanf("%d",&h[i]),maxsum=max(maxsum,h[i]);
        printf("Case #%d:\n",u++);
        if(maxsum>(n*m+1)/2)
        {
            printf("NO\n");
            continue;
        }
        flag=false;
        printf("YES\n");
        dfs(1,1,0);
    }
    return 0;
}