poj1190生日蛋糕

#include<iostream>
#include<cstdio>
#include<cmath>
#include<string>
#include<cstring>
#include<algorithm>
#include<cctype>
#include<queue>

using namespace std;

#define INF 10000000
#define min(a,b) (a<b?a:b)
int n,m,best,minv[22],mins[22];

void Init()
{
	int i;
	minv[0]=0;
	mins[0]=0;
	for(i=1;i<22;i++)//从顶层向下计算出最小体积和表面积的可能值
	{
		//从顶层(即第1层)到第i层的最小体积minv[i]成立时第j层的半径和高度都为j
		minv[i]=minv[i-1]+i*i*i;
		mins[i]=mins[i-1]+2*i*i;
	}
}
//dep:搜索深度,从底层m层向上搜,r,h分别为该层的半径和高度
void DFS(int dep,int sumv,int sums,int r,int h)
{
	int i,j,maxh;
	if(dep==0)//搜索完成,则更新最小面积值
	{
		if(sumv==n && sums<best)
			best=sums;
		return;
	}
	//剪枝如上面所述
	if((sumv+minv[dep-1]>n)||(sums+mins[dep-1]>best)||(2*(n-sumv)/r+sums>=best))
		return;
	//按递减顺序枚举dep层蛋糕半径的每一个可能值,这里第dep层的半径最小值为dep
	for(i=r-1;i>=dep;i--) 
	{
		if(dep==m)//底面积作为外表面积的初始值(总的上表面积,以后只需计算侧面积)
			sums=i*i;
		//最大高度,即dep层蛋糕高度的上限,(n-sumv-minv[dep-1])表示第dep层最大的体积
		maxh=min((n-sumv-minv[dep-1])/(i*i),h-1);
		for(j=maxh;j>=dep;j--)//同理,第dep层的最小高度值为dep
			DFS(dep-1,sumv+j*i*i,sums+2*i*j,i,j);//递归搜索子状态
	}
}

int main()
{
	Init();
	while(~scanf("%d%d",&n,&m))
	{
		best=INF;
		DFS(m,0,0,n+1,n+1);
		if(best==INF)
			best=0;
		printf("%d\n",best);
	}
	return 0;
}