hdu5113Black And White dfs暴搜 +剪枝

来源：互联网发布：苹果为什么4g网络很慢编辑：程序博客网时间：2024/04/28 15:41

//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 ;}

0 0