hdu 5113 Black And White dfs剪枝

这个。。。减枝方案看了题解才发现是自己思维太僵。。。
如果剩下的某种颜色大于剩下的格子数量的1/2那必有两个格子颜色相同然后return
鸽巢原理

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
int am[50],n,m,k,ans[10][10],flag;
void dfs(int i,int j)
{
     if(flag==1)return;
     if(i==n+1&&j==1)
     {
        flag =  1;
        printf("YES\n");
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
             {
                printf("%d",ans[i][j]);
                if(j==m)printf("\n");
                else printf(" ");
             }
        return ;
     }
     for(int _=1;_<=k;_++)
        if(am[_]>((n*m-((i-1)*m+j)+2)/2))return ;
     for(int _=1;_<=k;_++)
     {
         if(am[_]>0&&_!=ans[i-1][j]&&_!=ans[i][j-1])
         {
             am[_]--;
             ans[i][j] = _;
             if(j==m)dfs(i+1,1);
             else dfs(i,j+1);
             am[_]++;
         }
     }
}
int main()
{
    int t,i1 = 1;
    scanf("%d",&t);
    while(t--)
    {
       scanf("%d%d%d",&n,&m,&k);
       for(int i=1;i<=k;i++)scanf("%d",&am[i]);
       flag = 0;
       printf("Case #%d:\n",i1);i1++;
       dfs(1,1);
       if(!flag)printf("NO\n");
    }
    return 0;
}