POJ 生日蛋糕 DFS+剪枝

题目链接

题目大意：一个多层生日蛋糕。要求下面一层一定要比上面的一层大而且高，求符合该条件的蛋糕，其中最小的表面积； 

解题思路：

要求出最小表面积且下面的要比上面的大，所以每一层的半径和高度最小值为该层的层数（从上到下数的层数），这样一一枚举每层的半径和高度。但是由于情况太多，所以需要剪枝来降低搜索的次数。

剪枝一：剩余的蛋糕体积小于0；

剪枝二：高度和半径小于最小值（也就是当前层数）---从上往下数；

剪枝三：剩余体积不能够满足制作当前层蛋糕的需求（最小半径和高度）；

剪枝四：剩下蛋糕的体积，按照 制作前 求得每层 的最大体积，剩下的m层按照最大制作用不完蛋糕；

剪枝五：当前枚举求得的表面积已经大于之前求得的最小表面积；

圆柱公式 :
体积V = πHR^2
侧面积S' = 2πRH 
底面积S = πR^2

#include <cstdio>
#include <iostream>
#include <cmath>
using namespace std;
int mins[30],minv[30],Area,minArea,N,M;
int maxF(int m,int r,int h)//求最大体积
{
    int maxv=0;
    for(int i=0;i<m;i++)
    {
        maxv += (r-i)*(r-i)*(h-i);
    }
    return maxv;
}
void dfs(int n,int m,int r,int h)
{
    if(m==0)
    {
        if(n)
            return ;
        else if(minArea>Area)
        {
            minArea=Area;
            return ;
        }
    }
    if(n<=0)//无剩余体积
        return ;
    if(h<m || r<m)//高度或半径小于最小值
        return ;
    if(minv[m]>n)//剩余体积不满足最小蛋糕需求
        return ;
    if(maxF(m,r,h) < n)//剩下m层按最大设计也用不完
        return ;
    if(mins[m]+Area>=minArea)//当前层的理论最小面积+当前面积  已经大于  之前求得的最小面积时结束
        return ;
    for(int i=r;i>=m;i--)//i是半径
    {
        if(m==M)//最后一层
        {
            Area = i * i ;
        }
        for(int j=h;j>=m;j--)//j是高度
        {
            Area += 2*i*j;
            dfs(n-i*i*j, m-1, i-1, j-1);//i和j对应之前的加一  所以这里减一
            Area -= 2*i*j;
        }

    }
}
int main()
{
    while(cin>>N>>M)
    {
        mins[0]=0;
        minv[0]=0;
        for(int i=1;i<=M;i++)//理论最小面积和体积
        {
            mins[i]=mins[i-1]+2*i*i;
            minv[i]=minv[i-1]+i*i*i;
        }
        int maxr=sqrt((N-minv[M-1])/M) +1;
        int maxh=(N-minv[M-1])/(M*M) +1;
        ///保证非负 在dfs里会进行减一 不影响
        minArea=99999999;
        Area=0;
        dfs(N,M,maxr,maxh);
        cout<<minArea<<endl;
    }
    return 0;
}