G - Black And White

 #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 ;

 }