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[][];
 int n,m,k,cas=; //行，列，数量，案例几
 int flag = ;    //判定dfs return条件

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

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

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

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