POJ-1190生日蛋糕（详细题解）深度优先搜索

本文主要是介绍POJ-1190生日蛋糕（详细题解）深度优先搜索，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

POJ-1190

题目描述：

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外，以上所有数据皆为正整数）

输入

有两行，第一行为N（N <= 10000），表示待制作的蛋糕的体积为Nπ；第二行为M(M <= 20)，表示蛋糕的层数为M。

输出

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

例子输入

100
2

例子输出

68

提示

圆柱公式
体积V = πR 2H
侧面积A’ = 2πRH
底面积A = πR 2

在这里插入图片描述

蛋糕的体积： $V=Nπ=πR_1^2H_1+πR_2^2H_2+...+πR_n^2H_n$

则: $N=R_1^2H_1+R_2^2H_2+...+R_n^2H_n$
其中 $R_1>R_2>R_3>...>R_n且H_1>H_2>H_3>...>H_n)$

蛋糕的表面积： $Q=Sπ=πR_1^2+2πR_1H_1+2πR_2H_2+...+2πR_nH_n$

则: $S=R_1^2+2R_1H_1+2R_2H_2+...+2R_nH_n$
其中 $R_1>R_2>R_3>...>R_n且H_1>H_2>H_3>...>H_n)$

解题思路：
深度优先搜索：
从底层开始搜索，当底层半径和高度确定时，则搭建剩余层蛋糕所需的最小体积（min_v）和最大体积(max_v)就可以确定了，因此，每当我们搭建完一层时，应该比较一下max_v,min_v和剩余可用体积（ $V-V_i$ ）此时有三种情况：
第一种情况、

如果搭建完剩余层所需的最小体积（min_v）比剩余可用体积（ $V-V_i,其中V_i$ 表示搭建底层蛋糕已经使用的体积）还要大，则说明题目所固定的体积不够用，此时应该减小底层固定的半径 $R_i和H_i$ ，直到它们满足第三种情况

第二种情况、

如果搭建完剩余蛋糕层所需最大体积（max_v）比剩余可用体积（ $V-V_i$ ）还要小的话，则说用题目所固定的体积用不完，此时应该增大底层固定的半径 $R_i和H_i$ ，直到它们满足第三种情况

第三种情况、

如果搭建完剩余蛋糕层所需的最小体积（min_v）和最大体积（max_v）满足（min_v $V-V_i)<=$ max_v），则说明从第一层至当前层所搭建的蛋糕层符合题目要求，继续向上层搭建

由题目给定的数值，我们无需考虑π值，则上述的体积可用直接用整数值表示，比如：
当前已用体积 $V_i=R_1^2H_1+R_2^2H_2+...R_i^2H_i$
当前剩余体积 $V-V_i=N-(R_1^2H_1+R_2^2H_2+...R_i^2H_i)=R_{i+1}^2H_{i+1}+R_{i+2}^2H_{i+2}+...R_n^2H_n$

1.首先确定第一层半径R和高度H的最大值最小值：
总共有M层蛋糕，且每一层的半径和高度都是整数 $R_1>R_2>R_3>...>R_n且H_1>H_2>H_3>...>H_n)$ ，则最高层的半径和高度至少为1，由此可以确定第一层的半径和高度皆不小于M;

由 $N=R_1^2H_1+R_2^2H_2+...+R_n^2H_n$ 得：

当 $M=1,H_1=1$ 时, $R_1$ 最大，且此时 $R_1=\sqrt{N}$ ,所以 $R_1$ 的取值范围为[M, $\sqrt{N}$ ] 当 $R_1$ 的值确定时， $H_1=\frac{N}{R_1^2}$ ,同时 $H_1>=M$

2.依次确定上一层的半径和高的范围

$M-i+1<=R_i<=R_{i-1}-1;$
$M-i+1<=H_i<=H_{i-1}-1;$

3.根据已确定的半径和高度判断是否有可能满足题目要求
即分析上述的三种情况，max_v 和 min_v 和 V 三者之间的关系，若不满足则不断改变当前 R 和 H 的值，直到满足为止

4.确定最小的表面积
1.将第一次符合题目要求¹的表面积（min_s）记录，然后在遍历其它情况，如果在深度优先搜索的过程中发现已搭建的蛋糕表面积大于之前记录的最小表面积（min_s）则这种情况无需继续遍历，此时应该改变R和H的值，再进行另一种情况的遍历。
2.如果确立一种新的情况则判断当前蛋糕变面积和之前确立的最小表面积的值进行比较，如果当前 $S <$ min_s,则交换它们的值，（即更新最小表面积）

5.当枚举完所有情况时返回最小表面积的值

代码如下（C++）：

//time:563MS
#include<iostream>
#include<cmath>
#include<string>
#include<iomanip>
#include<map>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
int N, M;//分别表示体积和层数
int min_s = 0x7fffffff;//初始化为最大值
inline int min_v(int m)//此函数用于计算搭建剩余蛋糕层所需的最小体积
{int v = 0;//将剩余层的半径设为最小值，即从最高层到当前的上一层分别为1,2,3,4...M-m//每一层高度H取值和R一样,才能保证V最小；for (int R = M - m; R >= 1; R--){int H = R;v += R * R * H;}return v;
}inline int max_v(int r, int h, int m)//此函数用于计算搭建剩余蛋糕层所需的最大体积
{int v = 0;//由于上层半径和高度必须必下层半径和高度小，所以每一层半径和高度的最大取值应是其下层的半径或高度减一for (int R = r, H = h; m <= M; R--, H--,m++){//只有将剩余的每一层的半径和高度设为最大值，才能保证求得的V最大v += R * R * H;}return v;
}
inline void dfs(int r, int h, int m, int s, int v)//深度优先搜索，其中m表示当前在搭建的层数，s表示已经搭建的蛋糕层的表面积,v表示已经使用的体积
{if (m == M && N == v && s < min_s){min_s = s;//更新最小表面积的数值return;}//如果预估蛋糕的最小面积大于前面所确立最小面积，则无需继续深索if (r == 0 || s + 2 * (N - v) / r > min_s)return;//如果预估搭建剩余层所需的最小体积大于剩余可用体积，则说明规定的体积不够用，返回if (min_v(m) > N - v)//第一种情况return;//如果预估搭建剩余层所需的最大体积小于剩余可用体积，则说明无法达到规定的体积，返回if (max_v(r, h, m) < N - v)//第二种情况return;//如果上述条件均不满足，则说明满足第三种情况，则继续深索for (int R = r - 1; R >= M - m; R--)//枚举下一层蛋糕的半径for (int H = h - 1; H >= M - m; H--)//枚举下一层蛋糕的高{//注意，此时的最小值不是M-m+1,因为对应的是下一层的半径和高，所以最小值应该是M-mdfs(R, H, m + 1, s + 2 * R * H, v + R * R * H);//深搜下一层}
}
int main()
{scanf_s("%d%d", &N, &M);for (int r1 = M; r1 * r1 <= N; r1++)//枚举所有可能的半径和高度for (int h1 = N / (r1 * r1); h1 >= M; h1--){int s = r1 * r1 + r1 * h1 * 2;//表示第一层的表面积int v = r1 * r1 * h1;//表示第一层使用的体积dfs(r1, h1, 1, s, v);//固定了第一层的半径和高度之后，往上层去探索}if (min_s == 0x7fffffff)cout << 0 << endl;elsecout << min_s << endl;return 0;
}