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层的半径和高度都为jminv[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层的半径最小值为depfor(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层的最小高度值为depDFS(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;}