Black And White （DFS 训练题）

          G - Black And White

=========================================================================================================================

数据范围很小，可以直接用DFS深搜出结果，，但如果不把数据 按升序排好序 就去进行深搜的话，会超时，优先让数量最多的放在前面

然后 如果有一个数字的数量  > （M×N+1）/2 则肯定 输出 NO；因为怎么也不可能不相邻；

设DFS（int y ，int x） 即对这个格子进行染色

=========================================================================================================================

代码：

#include <algorithm>
#include <cstdio>
using namespace std;

int Map[30][30];
int n,m,k,cas=1; //行，列，数量，案例几
int flag = 0;    //判定dfs return条件

struct Node{
  int num,s;  //颜色编号，该颜色数量
}tail[40];      
bool cmp(Node a,Node b) { return a.s>b.s;}

void print() //输出函数
{
    printf("Case #%d:
YES
",cas++);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
                  printf(j==m?"%d
":"%d ",Map[i][j]);
}

void dfs(int y,int x)
{
    for(int i =1;i<=k;++i)
        if(flag==1) return;  
        else if(tail[i].s)
        {
            if(Map[y-1][x]!=tail[i].num&&Map[y][x-1]!=tail[i].num)
            {
                --tail[i].s;           //改颜色数量-1；
                Map[y][x] = tail[i].num;
                if(y==n&&x==m&&flag!=1) 
                    {flag = 1;print();return;} //如果格子填充完了，则输出
                else if(x==m)                //如果这一行填充完了，跳转至下一行
                    dfs(y+1,1);
                else                      //否则前往这一行的下一个格子
                    dfs(y,x+1);
                ++tail[i].s;
            }
        }
}
int main()
{
    int T;
    scanf("%d",&T);

    for(int num = 1;num<=T;++num)
    {
        scanf("%d %d %d",&n,&m,&k);
        for(int i = 1;i<=k;++i)
        {
            scanf("%d",&tail[i].s);
            tail[i].num = i;
        }
        sort(tail+1,tail+k+1,cmp); 
        if(tail[1].s>(n*m+1)/2) {printf("Case #%d:
NO
",cas++); continue;}
        flag = 0;
        dfs(1,1);
    }
    return 0;
}