洛谷 P1731 [NOI1999]生日蛋糕 && POJ 1190 生日蛋糕

题目传送门(洛谷)  OR 题目传送门(POJ)

解题思路:

一道搜索题,暴力思路比较容易想出来,但是这道题不剪枝肯定会TLE.所以这道题难点在于如何剪枝.

1.如果当前状态答案已经比我们以前某个状态求出来的答案还要大,那么我们就没有必要搜下去,直接return.

2.如果有某个状态,在这之后假设所有答案都是最优,还比我们当前已经求出来的最小值大,那么哇哦们也没有必要搜下去,return;

3.如果在某个状态之后,所有层蛋糕都用最大体积,也无法达到答案体积,那么也没必要搜,return.

以上三条剪枝对于这道题来说已经够了.

如果还想再优化一下,那么可以求出每层蛋糕当前R和H范围,在范围之间枚举,这样做也会使程序快一点,对于这道题来说,没必要//就因为写这个剪枝花了我1个半小时,最后还没写对,直接去掉这条剪枝,AC.

AC代码:

 #include<cstdio>
 #include<cmath>
 #include<iostream>
 #include<math.h> 

 using namespace std;

 int n,m,ans = 0x7f7f7f7f,_min[],oo[];

 int find_max(int nn,int h,int r) {
     int vv = ;
     for(int i = ;i <= m - nn + ; i++)
         vv += (h - i) * (r - i) * (r - i);
     return vv;
 }

 inline void dfs(int deep,int h,int r,int v,int len) {
     if(deep == m + ) {
         if(v != ) return ;
         ans = min(ans,len);
         return ;
     }
     if(v < ) return ;
     int u = len;
     u += _min[deep];
     if(ans < u) return ;
     u = find_max(deep,h,r);
     if(v > u && deep != ) return;
     for(int i = ;i < r; i++) {
         for(int j = ;j < h; j++) {
             if(i * i * j > v) continue;
             if(a[deep][i][j]) continue;
             if(deep == ) len = i * i;
             dfs(deep + ,j,i,v - j * i * i,len +  * j * i);
         }
     }
 }

 inline void special() {
     for(int i = ;i <= sqrt(n); i++) {
         if(n % (i * i) != ) continue;
         int j = n / (i * i);
         int len = i * i + i *  * j;
         ans = min(ans,len);
     }
 }

 int main()
 {
     scanf("%d%d",&n,&m);
     for(int p = m;p >= ; p--)
         _min[p] = _min[p+] + (m - p + ) *  * (m - p + );
     if(m == ) special();
     else dfs(,,,n,);
     if(ans == 0x7f7f7f7f) printf("");
     else printf("%d",ans);
     return ;
 }