Syins写的dfs例题——生日蛋糕

生日蛋糕

时间限制: 1 Sec

内存限制: 128 MB

题目描述

二月二十九日是ykh的生日，ACM-HLJU为此要制作一个体积为Nπ的M层生日蛋糕，每层都是一个圆柱体。
设从下往上数第i(1 <= i <= M)层蛋糕是半径为Ri, 高度为Hi的圆柱。当i < M时，要求Ri > Ri+1且Hi > Hi+1。
由于要在蛋糕上抹奶油，为尽可能节约经费，我们希望蛋糕外表面（最下一层的下底面除外）的面积Q最小。
令Q = Sπ
请编程对给出的N和M，找出蛋糕的制作方案（适当的Ri和Hi的值），使S最小。
（除Q外，以上所有数据皆为正整数）

输入

有两行，第一行为N（N <= 20000），表示待制作的蛋糕的体积为Nπ；第二行为M(M <= 20)，表示蛋糕的层数为M。

输出

仅一行，是一个正整数S（若无解则S = 0）。

样例输入

100
2

样例输出

68
一看就知道可用dfs，时限是1秒，tle 的可能很大
所以要考虑折枝
1：如果当前体积加上剩余的最小体积大于n那么返回
2：如果当前表面积大于之前得到的最小表面积那么返回
3：如果当前体积加上剩下的最大体积也小于n，那么返回
可以试一下数据：
10000
14
如果没有卡时间，一般就不会tle了；

代码👇

#include<iostream>
#include<math.h>
using namespace std;
int n,m,jg;
void dfs(int s,int lr,int lh,int sumv,int p);//s是当前已经好了的层数（从下往上），lr，lh分别是上一层的半径和高度，sumv是现在的体积和，p是现在的表面积；
int main()
{
    while(cin>>n){
        cin>>m;
        jg=0;
        dfs(0,sqrt(n)+2,n+1,0,0);
        cout<<jg<<endl;
    }
    return 0;
}
void dfs(int s,int lr,int lh,int sumv,int p)
{
    int x,y;
    if((m-s)*(lr-1)*(lr-1)*(lh-1)+sumv<n&&s)return;//折枝3
    if(sumv>n)return;//当前体积大于n就返回（小折枝）
    if(p>jg&&jg)return;//折枝2
    if(s==m){
        if(sumv==n&&(p<jg||jg==0))jg=p;
        return;
    }
    for(int i=lr-1;i>=m-s;i--){//m-s这块属于折枝1
        for(int j=lh-1;j>=m-s;j--){//m-s这块属于折枝1
            x=i*i*j;
            y=2*i*j;
            if(s==0)y+=i*i;
            if(p+2*(n-sumv)/i>jg&&jg)continue;
        dfs(s+1,i,j,sumv+x,p+y);
        }
    }
}