[NOI1999] 生日蛋糕

[NOI1999] 生日蛋糕

题目描述

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

设从下往上数第 $i$（$1\le i\le M$）层蛋糕是半径为 $R_i$，高度为 $H_i$ 的圆柱。当 $i<M$ 时，要求 $R_i>R_{i+1}$ 且 $H_i>H_{i+1}$。

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

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

（除 $Q$ 外，以上所有数据皆为正整数）

输入格式

第一行为一个整数 $N$（$N\le 2\times 10^4$），表示待制作的蛋糕的体积为 $N\pi$。

第二行为 $M$（$M\le 15$），表示蛋糕的层数为 $M$。

输出格式

输出一个整数 $S$，若无解，输出 $0$。

样例 #1

样例输入 #1

100
2

样例输出 #1

68

暴力dfs

代码与测试

#include<bits/stdc++.h>
using namespace std;
int r[25],h[25],N,M,minn=1e6;
void dfs(int dep,int s,int v,int agor,int agoh){
	if (N<v) return;
	if (dep==0){
		if (v==N) minn=min(minn,s);
		return;
	}
	int imaxd=agor-1;int jmaxd=agoh-1;
	for (int i=imaxd;i>=dep;i--){
		for (int j=jmaxd;j>=dep;j--){
			int stmp=2*i*j,vtmp=i*i*j;
			if (dep==M) dfs(dep-1,s+stmp+i*i,v+vtmp,i,j);
			else dfs(dep-1,s+stmp,v+vtmp,i,j);
		}
	}
}
int main(){
	cin>>N>>M;
	int &tm=r[M+1];
	r[M+1]=sqrt(N)+1;
	h[M+1]=(N)/tm*tm;
	dfs(M,0,0,r[M+1],h[M+1]);
	cout<<minn;
	return 0;
}

分析

基本的dfs，按题目要求打即可，不含剪枝，TLE5个

void dfs(int dep,int s,int v,int agor,int agoh){
	if (N<v) return;
	if (dep==0){
		if (v==N) minn=min(minn,s);
		return;
	}
	int imaxd=agor-1;int jmaxd=agoh-1;
	for (int i=imaxd;i>=dep;i--){
		for (int j=jmaxd;j>=dep;j--){
			int stmp=2*i*j,vtmp=i*i*j;
			if (dep==M) dfs(dep-1,s+stmp+i*i,v+vtmp,i,j);
			else dfs(dep-1,s+stmp,v+vtmp,i,j);
		}
	}
}

正在搜索蛋糕在dep层，当前外表面面积s，当前体积为v，使用agor记录上一次选的半径，agoh记录上一次选的高度。
使用倒序枚举，提高程序效率。

剪枝

#include<bits/stdc++.h>
using namespace std;
int r[25],h[25],N,M,minn=0x3f3f3f,s_b[25],v_b[25];
void dfs(int dep,int s,int v,int agor,int agoh){
	if (v+v_b[dep]>N) return;
	if (s+s_b[dep]>minn) return;
	if (s>minn) return;
	if (2*(N-v)/agor+s>=minn) return;
	if (dep==0){
		if (v==N) minn=min(minn,s);
		return;
	}
	int tmp=sqrt(N-v);
	int imaxd=min(agor-1,tmp);int t_jmaxd=agoh-1;
	for (int i=imaxd;i>=dep;i--){
		int jmaxd=min(t_jmaxd,(N-v)/(i*i));
		for (int j=jmaxd;j>=dep;j--){
			int stmp=2*i*j,vtmp=i*i*j;
			if (dep==M) dfs(dep-1,s+stmp+i*i,v+vtmp,i,j);
			else dfs(dep-1,s+stmp,v+vtmp,i,j);
		}
	}
	return;
}
int main(){
	cin>>N>>M;
	for (int i=1;i<=M;i++){
		s_b[i]=2*i*i+s_b[i-1];
		v_b[i]=i*i+v_b[i-1];
	}
	int &tm=r[M+1];
	r[M+1]=sqrt(N)+1;
	h[M+1]=N/tm*tm;
	dfs(M,0,0,r[M+1],h[M+1]);
	if (minn==0x3f3f3f) cout<<-1;
	else cout<<minn;
	return 0;
}

if (v+v_b[dep]>N) return;
if (s+s_b[dep]>minn) return;

for (int i=1;i<=M;i++){
		s_b[i]=2*i*i+s_b[i-1];
		v_b[i]=i*i+v_b[i-1];
	}

预处理从上到下前i层的最小体积与表面积，在dfs中做剪枝
2.

int imaxd=min(agor-1,tmp);int t_jmaxd=agoh-1;
	for (int i=imaxd;i>=dep;i--){
		int jmaxd=min(t_jmaxd,(N-v)/(i*i));
		for (int j=jmaxd;j>=dep;j--){
			int stmp=2*i*j,vtmp=i*i*j;
			if (dep==M) dfs(dep-1,s+stmp+i*i,v+vtmp,i,j);
			else dfs(dep-1,s+stmp,v+vtmp,i,j);
		}
	}

由于圆柱体积公式：
$$V=\pi R^{2}H$$
可以得：
$$\pi R^{2}H=\pi (N-v)$$
那么上下界就是：
R ∈ [ d e p , min ⁡ { N − v , a g o r − 1 } ] R \in [dep,\min \{\sqrt{N-v},agor-1\} ] R∈[dep,min{N−v ​,agor−1}]
$$H\in [dep,\min \{(N-v)/R^2,agoh-1\}]$$