poj1190 生日蛋糕 搜索+剪枝

首先写出最裸的剪枝。

我们可以发现第i层最小的r和h都是i（从上到下）

这样枚举就可以变小一点。

剪枝1：

若现在蛋糕体积太大（即剩下的蛋糕都按最小规格体积都会超 sumv+minv【m-flor】>n）

或者表面积太大（即剩下的蛋糕都按最小规格体积都会超 sums+mins【m-flor】>当前ans）

那么就不用往下了。

剪枝2：

剩下的都按当前半径为r的规格放都会比当前ans大（即在极限状态下都无法得到更优解）

就不用往下了。

剪枝3：

对最大高度的限制，即min { ( 剩下的体积-接下来（m-flor-1）层所需最小体积 ) / (当前面积)，上一层的半径-1}

至于为何是min 可以按可行性理解。

#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <iostream>
using namespace std;
int n,m,ans;
int mins[50],minv[50];
void dfs(int flor,int r,int h,int sumv,int sums)
{
    if(flor==m)
    {
        if(sumv==n&&sums<ans)
        {
            ans=sums;
        }
        return ;
    }
    if(sums+mins[m-flor]>ans || sumv+minv[m-flor]>n)
        return ;
    if(sums+2*(n-sumv)/r>ans)
        return;
    for(int i=r-1;i>=m-flor;i--)
    {
        if(flor==0)
        {
            sums=i*i;
        }
        int maxh=min(h-1,(n-sumv-minv[m-flor-1])/(i*i));
        for(int j=maxh;j>=m-flor;j--)
        {
            dfs(flor+1,i,j,sumv+i*i*j,sums+2*i*j);
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=21;i++)
    {
        mins[i]=mins[i-1]+2*i*i;
        minv[i]=minv[i-1]+i*i*i;
    }
    ans=0x3f3f3f3f;
    dfs(0,n,n,0,0);
    printf("%d\n",ans);
    return 0;
}