北大ACM(POJ1020-Anniversary Cake)

Question：http://poj.org/problem?id=1020

问题点：DFS。

 Memory: 260K        Time: 47MS
 Language: C++        Result: Accepted

 #include <iostream>
 using namespace std;

 int mat[];//按列记录小块Cake填充状态
 int side[];//小块Cake边长
 bool visit[];//小块Cake是否已使用
 int width,cnt;//大块Cake边长 小块Cake数量
 bool flag;//是否能被填充标志
 int cmp(const void* a,const void* b)
 {
     return *(int*)b - *(int*)a;//降序排列
 }
 //状态更新：    pn为1时，从x列开始填充第idx块Cake，并判断是否能填充进去
 //            pn为-1时，从x列开始清除第idx块Cake
 bool update(int idx,int x,int pn)
 {
     int i,temp = mat[x];
     if(pn == )
     {
         //填充第idx块Cake后，长宽不能超过边界
         if(side[idx] + x > width || mat[x] + side[idx] > width) return false;
         //判断x列到x+side[idx]-1列 是否同高度
         for(i=;i<side[idx];i++)
         {
             if(temp != mat[x+i]) return false;
         }
         visit[idx] = true;
     }
     else
     {
         visit[idx] = false;
     }
     for(i=;i<side[idx];i++)
     {
         mat[x+i] += side[idx]*pn;
     }
     return true;
 }
 //获取当前最小高度处的最小列，填充顺序为"从低到高，从左到右"
 int getLowX()
 {
     int i,temp=;
     for(i=;i<width;i++)
     {
         temp =(temp<mat[i]?temp:mat[i]);
     }
     for(i=;i<width && temp!= mat[i];i++);
     return i;
 }
 //idx为已填充数 ，x为小块Cake编号
 void dfs(int idx,int x)
 {
     if(idx == cnt) {//当全部填充完毕时，返回true
         flag = true;
         return;
     }
     for(int i=,w=;i<cnt;i++)
     {
         if(visit[i]) continue;//已填充的不再填充
         if(w == side[i]) continue; else w = side[i];//同边长的不重复填充
         if(update(i,x,))//在x列填充第i块Cake
         {
             dfs(idx+,getLowX());
             if(flag) return;
             update(i,x,-);//在x列清除第i块Cake
         }
     }
 }
 int main()
 {
     int eg;
     cin>>eg;
     while(eg--)
     {
         int i,j,k;
         memset(mat,,sizeof(mat));
         memset(side,,sizeof(side));
         memset(visit,,sizeof(visit));
         flag = false;
         cin>>width;
         cin>>cnt;
         for(i=;i<cnt;i++)
         {
             cin>>side[i];
         }
         qsort(side,cnt,sizeof(int),cmp);//排序用于去重
         dfs(,);
         if(flag) cout<<"KHOOOOB!"<<endl;
         else cout<<"HUTUTU!"<<endl;
     }
     //system("pause");
     return ;
 }