DFS+多剪枝
题目链接:http://poj.org/problem?id=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外,以上所有数据皆为正整数)
Input
有两行,第一行为N(N <= 10000),表示待制作的蛋糕的体积为Nπ;第二行为M(M <= 20),表示蛋糕的层数为M。
Output
仅一行,是一个正整数S(若无解则S = 0)。
Sample Input
100
2
Sample Output
68
Hint
圆柱公式
体积V = πR 2H
侧面积A’ = 2πRH
底面积A = πR 2
思路:
一直TLE,看了书上的剪枝才做出来.
可行性剪枝+最优剪枝都要好好考虑,尽量缩小搜索边界
具体剪枝在注释里写了
#include<iostream>
#include<cstdio>
#include<cmath>
#include<queue>
#include<vector>
#include<string>
#include<cstring>
#include<stack>
#include<algorithm>
using namespace std;
typedef long long ll;
int minarea=1<<30;
bool Is = false;
int area=0;
int N,M;
int minv[30]={0};// i层最小体积
int mina[30]={0};// i层最小侧面积
int maxrh(int m,int r,int h)
{
int v=0;
for(int i=0;i<m;i++)
v += (r-i)*(r-i)*(r-i);
return v;
}
void dfs(int n,int m,int r,int h)
{
if(n==0){
if(m==0){
Is = true;
minarea = min(minarea,area);
return ;
}
else {
return ;
}
}
if(n<=0) return ; // 搭建失败
if(minv[m]>n) return; // 当前体积不够搭完
if(area+mina[m]>=minarea) return ;// 当前策略一定不是最优解
if(h<m||r<m) return ; // 高度与半径都无法递减到结束
if(maxrh(m,r,h)<n) return ;// 当前体积太多,用不完
for(int rr = r;rr >= m;rr--){
if(m==M) area = rr * rr;
for(int hh = h;hh >= m;hh--){
area += 2 * rr * hh;
dfs(n-rr*rr*hh,m-1,rr-1,hh-1);
area -= 2 * rr * hh;
}
}
}
int main()
{
cin >> N >> M;
for(int i=1;i<=M;i++){ // 第i层半径最少为i,高度最少为i
minv[i]= minv[i-1]+i*i*i;
mina[i]= mina[i-1]+2*i*i;
}
int maxH = (N-minv[M-1])/(M*M)+1;
int maxR = sqrt(double((N-minv[M-1])/M+1));
dfs(N,M,maxR,maxH);
if(Is)
cout << minarea;
else
cout << "0";
return 0;
}