POJ 1190 生日蛋糕题解

题目地址：http://poj.org/problem?id=1190

一道很有趣的搜索题……主要是剪枝……

我弄了5个剪枝：

1.当前剩余层数>=上层半径，剪掉

2.当前剩余层数>=上层高度，剪掉

3.体积<1，剪掉

4.体积<当前可能的最小体积（我用等差数列求和公式算的……也不知道对不对）

5.神奇剪枝网上看的，当前(剩余体积*2)/上层半径+当前表面积总和>当前搜出的最优解，剪掉

代码很简单，20行

#include <iostream>
#include <cmath>
#include <climits>
#define pow2(k) k*k
#define v(r,h) pow2(r)*h
#define min(a,b) a<b?a:b
using namespace std;
int n,m,s=INT_MAX,beg;
void dfs(int r,int h,int c,int lv,int ans)
{
    if(c>m){if(!lv)s=min(s,ans);return;}if(r<=m-c+1||h<=m-c+1||ans+lv*2/r>s||lv<1||lv<(m-c+2)*(m-c+1)/2)return;
    for(int i=r-1;i>0;i--)for(int j=1;j<h&&v(i,j)<=lv;j++)dfs(i,j,c+1,lv-v(i,j),ans+2*i*j);
}
int main()
{
    ios::sync_with_stdio(0);
    cin>>n>>m,beg=sqrt(n)+1;
    for(int i=beg;i>=1;i--)for(int j=1;v(i,j)<=n;j++)dfs(i,j,2,n-v(i,j),pow2(i)+2*i*j);
    cout<<(s==INT_MAX?0:s);
    return 0;
}