生日蛋糕 POJ - 1190--DFS+多剪枝

DFS+多剪枝
题目链接:http://poj.org/problem?id=1190
生日蛋糕
7月17日是Mr.W的生日，ACM-THU为此要制作一个体积为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外，以上所有数据皆为正整数）
Input
有两行，第一行为N（N <= 10000），表示待制作的蛋糕的体积为Nπ；第二行为M(M <= 20)，表示蛋糕的层数为M。
Output
仅一行，是一个正整数S（若无解则S = 0）。
Sample Input
100
2
Sample Output
68
Hint
圆柱公式
体积V = πR 2H
侧面积A’ = 2πRH
底面积A = πR 2
思路:
一直TLE,看了书上的剪枝才做出来.
可行性剪枝+最优剪枝都要好好考虑,尽量缩小搜索边界
具体剪枝在注释里写了

#include<iostream>
#include<cstdio>
#include<cmath>
#include<queue>
#include<vector>
#include<string>
#include<cstring>
#include<stack>
#include<algorithm>
using namespace std;
typedef long long ll;
int minarea=1<<30;
bool Is = false;
int area=0;
int N,M;
int minv[30]={0};// i层最小体积
int mina[30]={0};// i层最小侧面积
int maxrh(int m,int r,int h)
{
    int v=0;
    for(int i=0;i<m;i++)
        v += (r-i)*(r-i)*(r-i);
    return v;
}
void dfs(int n,int m,int r,int h)
{
    if(n==0){
        if(m==0){
            Is = true;
            minarea = min(minarea,area);
            return ;
        }
        else {
            return ;
        }
    }
    if(n<=0) return ; // 搭建失败
    if(minv[m]>n) return; // 当前体积不够搭完
    if(area+mina[m]>=minarea) return ;// 当前策略一定不是最优解
    if(h<m||r<m) return ; // 高度与半径都无法递减到结束
    if(maxrh(m,r,h)<n) return ;// 当前体积太多,用不完
    for(int rr = r;rr >= m;rr--){
        if(m==M) area = rr * rr;
        for(int hh = h;hh >= m;hh--){
            area += 2 * rr * hh;
            dfs(n-rr*rr*hh,m-1,rr-1,hh-1);
            area -= 2 * rr * hh;
        }
    }
}
int main()
{
    cin >> N >> M;
    for(int i=1;i<=M;i++){ // 第i层半径最少为i,高度最少为i
        minv[i]= minv[i-1]+i*i*i;
        mina[i]= mina[i-1]+2*i*i;
    }
    int maxH = (N-minv[M-1])/(M*M)+1;
    int maxR = sqrt(double((N-minv[M-1])/M+1));
    dfs(N,M,maxR,maxH);
    if(Is)
        cout << minarea;
    else
        cout << "0";
    return 0;
}