dfs-hdu 5113 Black And White

题意：

中文翻译：

有一个包含从1到K一共K种颜色的N×M棋盘，使得任意两个相邻的区块不能有相同的颜色（如果它们的上、下、左、右任意一边的颜色与自身颜色不同）。第i种颜色可以被使用Ci次。

Input

第一行包含一个整数T（1<=T<= 5000）, T表示测试用例的数量。

    对每一个测试用例第一行包含三个整数N,M,K(0<N,M<=5,0<K<=N*M)。

    第二行包含K个整数Ci(Ci>0),表示第i种颜色可以被使用的次数。

    输入保证C1+ C2+...+ CK=N*M。

Output

对每个测试用例，第一行包含“Case #x:”, x是用例的序号( 从1开始)。

    如果有满足条件的涂色方法就输出“YES”，如果没有就输出“NO”。接下来的N行每行由M个表示第i种颜色的数字构成，数字之间有一个空格隔开。

    如果有多重涂法，输出其中的一种。

思路：

从左上到右下，逐次搜索，注意上左同这个颜色不能相等，且剩余位置的一半如果小于某种颜色就return

#include <iostream>
#include <stdio.h>
#include <cstring>
using namespace std;
int col[50];
int mp[50][50];
int n,m,k;
int yes;
void dfs(int x,int y,int res)
{
    if(res==0||yes )
    {
        yes=1;
        return ;
    }
    for(int i=1;i<=k;i++)
    {
        if((res+1)/2<col[i])
        return ;
    }
    for(int i=1;i<=k;i++)
    {
        if(col[i]!=0)
        {
            if(mp[x][y-1]!=i&&mp[x-1][y]!=i)
            {
                mp[x][y]=i;
                col[i]--;
                if(y==m)
                {
                    dfs(x+1,1,res-1);
                }
                else
                {
                    dfs(x,y+1,res-1);
                }
                if(yes)
                return ;
                mp[x][y]=0;
                col[i]++;
            }
        }
    }
}
int main()
{
    int T;
    cin>>T;
    for(int Case=1;Case<=T;Case++)
    {
        memset(mp,0,sizeof(mp));
        yes=0;
        scanf("%d%d%d",&n,&m,&k);
        for(int i=1; i<=k; i++)
            scanf("%d",&col[i]);
        dfs( 1,1,n*m);

        printf("Case #%d:\n",Case);
        if(yes)
        {
            printf("YES\n");
            for(int i=1; i<=n; i++)
            {
                printf("%d",mp[i][1]);
                for(int j=2; j<=m; j++)
                    printf(" %d",mp[i][j]);
                printf("\n");
            }
        }
        else
            printf("NO\n");
    }
    return 0;
}
/*
4
2 2 2
2 2

*/