hdu 5113 Black And White

来源：互联网发布：茂名行知中学以前是编辑：程序博客网时间：2024/05/20 08:24

题意：有一个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;}

0 0