hdu5113Black And White dfs暴搜 +剪枝

//n(n<=5)*m(m<=5)的格子
//k种颜料，每种颜料有ai个
//问这k种颜料能否将这n*m涂满且
//每相邻的格子颜色不同
//暴搜 , 剪枝剩下的颜色的数量大于剩下(sum+1)/2
//直接return
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std ;
const int maxn = 30 ;
int map[maxn][maxn] ;
int a[maxn] ;
int n , m , k ;
bool dfs(int x , int y , int sum)
{
    y++ ;
    if(y == m){
       x++ ;
       y = 0 ;
    }
    if(x ==n){
       return true ;
    }
    for(int i = 1;i <= k;i++)
    if(a[i] > (sum+1)/2)return false ; 
    for(int i = 1;i <= k;i++)
    if(a[i]){
       int px = x - 1 ;
       int py = y - 1 ;
       if(px >= 0 && map[px][y] == i)
       continue ;
       if(py >=0 && map[x][py] == i)
       continue ;
       map[x][y] = i ;
       a[i]-- ;
       if(dfs(x , y , sum-1))return true ;
       a[i]++ ;
    }
    return false  ;
}
int main()
{
    int t ;
    int cas = 0 ;
    scanf("%d" , &t) ;
    while(t--)
    {
        scanf("%d%d%d" , &n , &m , &k) ;
        for(int i = 1;i <= k;i++)
        scanf("%d"  , &a[i]) ;
        printf("Case #%d:\n" , ++cas);
        if(dfs(0 , -1 , n*m))
        {
            puts("YES") ;
            for(int i = 0;i < n;i++)
               for(int j = 0;j < m;j++)
                 printf("%d%c" , map[i][j] , j == m - 1?'\n':' ') ;
        }
        else puts("NO") ;
    }
    return 0 ;
}