hdu5113 dfs+强有力剪枝（将颜色放满格子）

注意到，只要在搜索过程中出现一种颜色大于剩下格子/2，必然不用继续向下搜索。

 1 #include<stdio.h>
 2 #include<string.h>
 3 #include<algorithm>
 4 using namespace std;
 5 int d[105][105],c[105];
 6 int n,m,k,temp;
 7 int judge(int x,int y,int i)
 8 {
 9     if (y==1&&x==1) return 1;
10     if (x==1)
11         if (i==d[x][y-1]) return 0;
12         else return 1;
13     if (y==1)
14         if (i==d[x-1][y]) return 0;
15         else return 1;
16     if (i==d[x-1][y]||i==d[x][y-1]) return 0;
17     else return 1;
18 }
19 void dfs(int x,int y)
20 {
21     int i,j;
22     if (temp) return;
23     int step=(n-x)*m+m-y;
24     for (i=1;i<=k;i++)
25         if (c[i]>step/2+1) return;
26     if (x>n){
27         temp=1;
28         printf("YES\n");
29         for (i=1;i<=n;i++)
30         {
31             for (j=1;j<m;j++)
32                 printf("%d ",d[i][j]);
33             printf("%d\n",d[i][m]);
34         }
35         return;
36     }
37     for (i=1;i<=k;i++)
38         if (c[i]&&judge(x,y,i))
39     {
40         c[i]--;
41         d[x][y]=i;
42         if (y==m) dfs(x+1,1);
43         else dfs(x,y+1);
44         c[i]++;
45         d[x][y]=-1;
46     }
47     return;
48 }
49 int main()
50 {
51     int T,t,i;
52     scanf("%d",&T);
53     for (t=1;t<=T;t++)
54     {
55         scanf("%d%d%d",&n,&m,&k);
56         for (i=1;i<=k;i++)
57             scanf("%d",&c[i]);
58         memset(d,-1,sizeof(d));
59         printf("Case #%d:\n",t);
60         temp=0;  dfs(1,1);
61         if (!temp) printf("NO\n");
62     }
63 }

View Code

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5113

转载于:https://www.cnblogs.com/xiao-xin/articles/4133096.html