POJ1020 Anniversary Cake

 1 #include<iostream>
 2 #include<algorithm>
 3 #include<cstring>
 4 #include<cstdio>
 5 using namespace std;
 6 int T,n,m;
 7 int a[11];//边长为[i]的蛋糕的数量 
 8 int ck[50];//第[i]列已填的高度 
 9 bool dfs(int y){
10     if(y==m)return 1;
11     int i,j;
12     //
13     int mi=200,pos=0;
14     for(j=1;j<=n;j++)
15         if(ck[j]<mi){
16             mi=ck[j];
17             pos=j;    
18         }
19     //
20     for(i=10;i;i--){
21         if(!a[i])continue;//没有该尺寸蛋糕，跳过 
22         if(pos+i-1>n)continue;//该尺寸宽度比剩余宽度大，跳过 
23         for(j=0;j<i;j++){
24             if(ck[pos+j]<=ck[pos] && ck[pos+j]+i<=n)continue;
25             else break;
26         }
27         if(j<i)continue;//高度不足，跳过 
28         for(j=0;j<i;j++)ck[pos+j]+=i;
29         a[i]--;
30         if(dfs(y+1))return 1;
31         a[i]++;
32         for(j=0;j<i;j++)ck[pos+j]-=i;
33     }
34     return 0;
35 }
36 int main(){
37     scanf("%d",&T);
38     int i,j;
39     while(T--){
40         memset(ck,0,sizeof ck);
41         memset(a,0,sizeof a);
42         scanf("%d%d",&n,&m);
43         int x;
44         int ssum=0;
45         int cnt=0;
46         for(i=1;i<=m;i++){
47             scanf("%d",&x);
48             a[x]++;
49             ssum+=x*x;
50             if(x>n/2)cnt++;//尺寸大于n/2的蛋糕若有多块，肯定不可行 
51         }
52         if(ssum!=n*n || cnt>1){
53             printf("HUTUTU!\n");
54             continue;
55         }
56         if(!dfs(0)){
57             printf("HUTUTU!\n");
58         }
59         else printf("KHOOOOB!\n");
60         
61     }
62     return 0;
63 }