【POJ - 1190】生日蛋糕 （dfs+剪枝）

#include <iostream>
#include <cstdio>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <deque>
#include <vector>
#include <queue>
#include <string>
#include <cstring>
#include <map>
#include <stack>
#include <set>
#include <sstream>
#define mod 1000000007
#define eps 1e-6
#define ll long long
#define INF 0x3f3f3f3f
#define MEM(x,y) memset(x,y,sizeof(x))
#define Maxn 25
using namespace std;
int V,M;
int minArea;
int area;
int minv[Maxn],minarea[Maxn];
void init()
{
    minv[0]=0;
    minarea[0]=0;
    for(int i=1; i<Maxn; i++)
    {
        minv[i]=minv[i-1]+i*i*i;
        minarea[i]=minarea[i-1]+2*i*i;
    }
}
//剩余层数也是第m层、当前所有体积、当前所有面积，当前这层蛋糕的半径，当前这层蛋糕的高
void dfs(int m,int v,int area,int r,int h)
{
    if(m==0)//搜索完成,则更新最小面积值
    {
        if(v==V&&area<minArea)
            minArea=area;
        return;
    }
    //剪枝
    if(minv[m-1]>V-v||2*(V-v)/r+area>=minArea)
        return;
    //按递减顺序枚举m层蛋糕半径的每一个可能值
    for(int i=r-1; i>=m; i--)
    {
        if(m==M)//底面积作为外表面积的初始值
            area=i*i;
        //最大高度,即m层蛋糕高度的上限
        int maxh=min(h-1,V-area-minarea[m-1]/(i*i));
        for(int j=maxh; j>=m; j--)
        {
            dfs(m-1,v+i*i*j,area+2*i*j,i,j);
        }
    }
}
int main()
{
    init();
    minArea=INF;
    cin>>V>>M;//面积，层数
    dfs(M,0,0,V+1,V+1);
    if(minArea==INF)
        cout<<0<<endl;
    else
        cout<<minArea<<endl;
}