『算法-ACM 竞赛-DFS』POJ 1190 生日蛋糕

Description
7 月 17 日是 Mr.W 的生日，ACM-THU 为此要制作一个体积为 Nπ 的 M 层生日蛋糕，每层都是一个圆柱体。
设从下往上数第 i(1 <= i <= M)层蛋糕是半径为 Ri, 高度为 Hi 的圆柱。当 i < M 时，要求 Ri > Ri+1 且 Hi > Hi+1。
由于要在蛋糕上抹奶油，为尽可能节约经费，我们希望蛋糕外表面（最下一层的下底面除外）的面积 Q 最小。
令 Q = Sπ
请编程对给出的 N 和 M，找出蛋糕的制作方案（适当的 Ri 和 Hi 的值），使 S 最小。
（除 Q 外，以上所有数据皆为正整数）
Input
有两行，第一行为 N（N <= 10000），表示待制作的蛋糕的体积为 Nπ；第二行为 M(M <= 20)，表示蛋糕的层数为 M。
Output
仅一行，是一个正整数 S（若无解则 S = 0）。
Sample Input
100
2
Sample Output
68
Solution
由于深度一定(m),所以使用深度优先搜索,自上而下的设定蛋糕序号,最顶层的为第 1 层,……,最底层的蛋糕为第 m 层,很明显满足题目条件的前 i 层的(从顶层(也就是编号为 1 的层)开始计数)最小面积 mins[i]和体积 minv[i]是在该层的半径以及高度都为 i 时取得,如果采用一般的神搜肯定会超时,所以这题还需要剪枝,剪枝条件有(从 m 层向上搜,假设前 level 层的体积为 v,面积为 s,当前所得的最小面积为 best):
1>因为前 level 层的体积为 v,如果剩下的几层的体积都取最小可能值,总体积还是比 n 大,那么则说明前 level 层的方案不可行,所以可以剪枝(剪枝条件为:v+minv[dep-1]>n)
2>因为前 level 层的面积为 s,如果剩下的几层的面积都取最小可能值,所得的面积和比已经得到的所求的最小面积 best 大,也可以进行剪枝(剪枝条件为:s+mins[dep-1]>best)
3>因为前 level 层的体积为 v,那么剩余的 m-level 层的体积满足:n-v=(hk+……+hm)(k=level+1,……,m)
而剩余部分的表面积满足:lefts=2*(r[k]h[k]+……+r[m]h[m])>2(n-sv)/r[level] (k=level+1,……,m)
显然有上述不等式 lefts=best-s>2(n-)/r,即 2*(n-v)/r+s

#include<stdio.h>
#define in(a,b) (a<b?a:b)
int n,m;
int minv[21],mins[21];
int bests;

void dfs(int v,int s,int level,int r,int h)//level为搜索深度,从底层m层向上搜,r,h分别为该层的半径和高度
{
    if(level==0)//搜索完成,则更新最小面积值
    {
        if(v==n&&s<bests)
            bests=s;
        return ;
    }
    if(v+minv[level-1]>n||s+mins[level-1]>bests||2*(n-v)/r+s>=bests)//剪枝
        return ;
    int i,j,hh;
    for(i=r-1;i>=level;i--)//按递减顺序枚举level层蛋糕半径的每一个可能值,这里第level层的半径最小值为level
    {
        if(level==m)//底面积作为外表面积的初始值(总的上表面积,以后只需计算侧面积)
            s=i*i;
        hh=in((n-v-minv[level-1])/(i*i),h-1); //最大高度,即level层蛋糕高度的上限,(n-v-minv[level-1])表示第level层最大的体积
        for(j=hh;j>=level;j--)//同理,第level层的最小高度值为level
            dfs(v+i*i*j,s+2*i*j,level-1,i,j);//递归搜索子状态
    }
}
int main()
{
    int i;
    minv[0]=0;
    mins[0]=0;
    for(i=1;i<=20;i++)//从顶层向下计算出最小体积和表面积的可能值
    {
        //从顶层（即第一层）到第i层的最小体积minv[i]成立时第j层的半径和高度都是j
        minv[i]=minv[i-1]+i*i*i;
        mins[i]=mins[i-1]+2*i*i;
    }
    while(scanf("%d%d",&n,&m)==2)
    {
        bests=0x7fffffff;
        dfs(0,0,m,n+1,n+1);
        if(bests==0x7fffffff)
            printf("0\n");
        else
            printf("%d\n",bests);
    }
    return 0;
}