hdu 5113 Black And White(暴力+剪枝)

题目链接：hdu 5113 Black And White

题意：

给你n*m的格子，现在有k种颜色，每种颜色要填c[i]个格子，让你输出一种合法的填涂方案，相邻格子颜色不能相同。

题解：

暴搜+剪枝。

对于当前剩下的格子数，考虑剩下的c[i]，如果剩下的c[i]>(剩下的格子数+1)/2，那么就不用往下搜了。

这样速度飞快，46msAC。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;

int t,n,m,k,cas,a[30],flag,mp[10][10];

int check(int have)
{
    F(i,1,k)if(a[i]>(have+1)/2)return 1;
    return 0;
}

void dfs(int now,int have)
{
    if(flag)return;
    if(now>n*m)
    {
        int is=1;
        F(i,1,k)if(a[i]){is=0;break;}
        if(is)
        {
            puts("YES"),flag=1;
            F(i,1,n)F(j,1,m)printf("%d%c",mp[i][j]," \n"[j==m]);
            return;
        }
    }
    if(check(have))return;
    int x=(now-1)/m+1,y=(now-1)%m+1;
    F(i,1,k)if(a[i])
    {
        if(mp[x-1][y]==i||mp[x][y-1]==i)continue;
        mp[x][y]=i;
        a[i]--;
        dfs(now+1,have-1);
        a[i]++;
    }
}

int main(){
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&n,&m,&k);
        F(i,1,k)scanf("%d",a+i);
        printf("Case #%d:\n",++cas);
        flag=0,dfs(1,n*m);
        if(!flag)puts("NO");
    }
    return 0;
}