DFS(剪枝) - POJ 1190 - NOI1999 生日蛋糕

DFS(剪枝) - POJ 1190 - NOI1999 生日蛋糕

7月17日是 Mr.W 的生日，ACM-THU 为此要制作一个体积为 $N\pi$ 的 M 层生日蛋糕，每层都是一个圆柱体。

设从下往上数第i层蛋糕是半径为$R_i$, 高度为$H_i$的圆柱。

当 i < M 时，要求$R_i>R_{i+1}$且$H_i>H_{i+1}$。

由于要在蛋糕上抹奶油，为尽可能节约经费，我们希望蛋糕外表面（最下一层的下底面除外）的面积 Q 最小。

令$Q=S\pi$ ，请编程对给出的 N 和 M ，找出蛋糕的制作方案（适当的$R_i$和$H_i$的值），使 S 最小。

除 Q 外，以上所有数据皆为正整数 。

输入格式
输入包含两行，第一行为整数 N（N <= 10000），表示待制作的蛋糕的体积为 $N\pi$。

第二行为整数 M (M <= 20) ，表示蛋糕的层数为 M。

输出格式
输出仅一行，是一个正整数 S（若无解则S = 0）。

数据范围
$1\le N\le 10000,1\le M\le 20$

输入样例：

100
2

输出样例：

68

---

分析：

目标： $给定总体积为N，总层数为M的蛋糕，求表面积的最小值(除下底面)。$

$半径为R_u，高为H_u的圆柱侧面积：S_u=2\pi R_u{H}_u；体积V_u=\pi R_u^2{H}_u。$

$本题不考虑\pi ，S_u=2R_u{H}_u；体积V_u=R_u^2{H}_u。$

$我们自顶向下给每一层编号为1，2，...，M。$

$$\begin{gathered} 根据R和H的递增关系，dfs枚举每一层的R和H的值， \\ 判断M层蛋糕体积是否恰好为N，求合法方案中表面积的最小值。 \end{gathered}$$

剪枝：

$$\begin{gathered} ①、最底层的蛋糕半径最大，体积最大，能够减少枚举体积的分支，因此我们自底向上搜索。 \\ 由于R对体积的影响要大于H，所以我们优先枚举R。 \end{gathered}$$

$$\begin{gathered} ②、由于自顶向下每一层的半径和高是递增的，且R和H是整数，故对第u层圆柱，有不等关系： \\ u\le R_u\le R_{u+1}-1 \\ u\le H_u\le H_{u+1}-1 \end{gathered}$$

$$\begin{gathered} 假设第u层以下的蛋糕总体积为v，则剩下的体积总量为N-v， \\ 则有N-v\ge R_u^2{H}_u\ge R_u^2，得到R_u\le \sqrt{N-v}。 \end{gathered}$$

$同理由N-v\ge R_u^2{H}_u，得到H_u\le \frac{N-v}{R_u^2}。$

$综上：u\le R_u\le min(R_{u+1}-1,\sqrt{N-v})，u\le H_u\le min(H_{u+1}-1,\frac{N-v}{R_u^2})。$

$$\begin{gathered} ③、预处理出前u层体积最小值minv[u]和mins[u]。则对第u层蛋糕的体积V_u和表面积S_u，有 \\ {V}_u+minv[u]\le N，S_u+mins[u]<ans。其中ans是当前已经枚举到的合法方案的表面积的最小值。 \end{gathered}$$

$预处理方案就是半径按照1,2,...,M递增，即R_u=H_u=u，u\in [1,M]。$

$可见，对V_u是可行性剪枝，对S_u是最优性剪枝。$

$④、1到u层的表面积之和S_{1-u}={\sum }_{i=1}^{u}2R_i{H}_i。$

$由于我们是自底向上枚举，设第u层以下已经枚举过的蛋糕总体积为v，则N-v=V_{1-u}={\sum }_{i=1}^u{R}_i^2{H}_i。$

$则有S_{1-u}={\sum }_{i=1}^{u}2R_i{H}_i=2{\sum }_{i=1}^u{R}_i{H}_i{R}_{i+1}\cdot \frac{1}{R_{i+1}}=\frac{2}{R_{i+1}}{\sum }_{i=1}^u{R}_i{H}_i{R}_{i+1}＞\frac{2}{R_{i+1}}{\sum }_{i=1}^u{R}_i^2{H}_i=\frac{2}{R_{i+1}}{V}_{1-u}。$

$即S_{1-u}>\frac{2}{R_{i+1}}{V}_{1-u}=\frac{2(N-v)}{R_{i+1}}。$

$因此，设第u层以下已经枚举过的蛋糕总的表面积为s，若s+\frac{2(N-v)}{R_{i+1}}\ge ans，说明该分支不能更新最优解。$

$其中ans是当前已经枚举到的合法方案的表面积的最小值。这里仍然是最优性剪枝。$

代码：

#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>

using namespace std;

const int N=25,inf=1e9;

int n,m;
int minv[N],mins[N];
int R[N],H[N];
int ans=inf;

void dfs(int u,int v,int s)
{
    if(v + minv[u] > n) return;  //可行性剪枝
    if(s + mins[u] >= ans) return;  //最优性剪枝
    if(s + 2*(n-v)/R[u+1] >= ans) return;  //最优性剪枝
    
    if(u==0)   //到顶
    {
        if(v==n) ans=s;
        return;
    }
    
    for(int r = min(R[u+1]-1,(int)sqrt(n-v)); r >= u; r--)  //从大到小枚举
        for(int h = min(H[u+1]-1,(n-v)/r/r); h >= u; h--)
        {
            int tmp=0;
            if(u==m) tmp=r*r;  //最底层先加上上表面面积
            R[u]=r,H[u]=h;
            dfs(u-1,v+r*r*h,s+2*r*h+tmp);
        }
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        minv[i]=minv[i-1]+i*i*i;
        mins[i]=mins[i-1]+2*i*i;
    }
    
    R[m+1]=H[m+1]=inf;  //最后一层要用到下一层的参数，故初始化边界为inf
    
    dfs(m,0,0);
    
    if(ans==inf) ans=0;  //无解输出0
    cout<<ans<<endl;
    
    return 0;
}

---