POJ 1190 生日蛋糕

生日蛋糕

#include<iostream>
#include<cstdio>
#include<queue>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=10000+5;
int N,M,minS,minV[maxn],minA[maxn];
int getMaxS(int r,int h,int H)
{
	int ans=0;
	for(int i=0;i<H;i++)
	ans+=(h-i)*(r-i)*(r-i);
	return ans;
}
int dfs(int V,int H,int r,int h,int S)
{   // 要达成的体积，位于第几层，此时半径，高度，侧面积 
	if(!H){
		if(V) return 0;
		minS=min(minS,S);
		return 0;
	}
	if(V<=0) return 0;
	if(minV[H]>V) return 0;
	if(S+minA[H]>minS) return 0;
	if(h<H||r<H) return 0;
	if(getMaxS(r,h,H)<V) return 2;
	
	for(int i=r;i>=H;i--)
	{
		if(H==M) S=i*i;
		for(int j=h;j>=H;j--)
			if(dfs(V-i*i*j,H-1,i-1,j-1,S+2*i*j)) break;
	}
	return 0;
}
int main()
{
	cin>>N>>M;
	for(int i=1;i<=M;i++){
		minV[i]=minV[i-1]+i*i*i;
		minA[i]=minA[i-1]+2*i*i;
	}
	if(N<minV[M]) cout<<0<<endl;
	else {
		minS=(1<<30);
		dfs(N,M,(N-minV[M-1])/(M*M)+1,(int)sqrt((double)(N-minV[M-1])/M)+1,0);
		if(minS==(1<<30)) cout<<0<<endl;
		else cout<<minS<<endl;
	}
	return 0;
	
}