洛谷P1731 [NOI1999]生日蛋糕【搜索+剪枝】

题目描述 https://www.luogu.org/problemnew/show/P1731
先预处理出表面积和体积的最优解，方便后面的剪枝，上表面的面积就是底面圆的面积，枚举半径和高的时候，要注意上下的范围。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,m,ms[25],mv[25],ans=9999999; 
void dfs(int s,int v,int p,int r,int h)//m层为最底层 
{//p+1层表面积 p+1层体积 第p层 p+1层的半径 高
    if(p==0)
    {
    	if(v==n&&ans>s) ans=s;
    	return;
    }
	if(s+ms[p-1]>ans) return;
	if(v+mv[p-1]>n) return;
	if(2*(n-v)/r+s>=ans) return;
	for(int i=r-1;i>=p;i--)//枚举半径 
	{
		if(p==m) s=i*i;
		int t=min((n-v-mv[p-1])/(i*i),h-1);
		for(int j=t;j>=p;j--)//枚举高 
		  dfs(s+2*i*j,v+i*i*j,p-1,i,j); 
	}
	
}
int main() 
{ 
   scanf("%d%d",&n,&m);
   for(int i=1;i<=m;i++)
   {
   	  ms[i]=ms[i-1]+2*i*i;
   	  mv[i]=mv[i-1]+i*i*i;
   }
   dfs(0,0,m,n+1,n+1);
   if(ans==9999999) puts("0");
   else printf("%d",ans);
 	
  return 0;	
}