首页 > 代码库 > POJ 1190 生日蛋糕题解

POJ 1190 生日蛋糕题解

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

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

我弄了5个剪枝:

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

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

3.体积<1,剪掉

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

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

 

代码很简单,20行

 1 #include <iostream>
 2 #include <cmath>
 3 #include <climits>
 4 #define pow2(k) k*k
 5 #define v(r,h) pow2(r)*h
 6 #define min(a,b) a<b?a:b
 7 using namespace std;
 8 int n,m,s=INT_MAX,beg;
 9 void dfs(int r,int h,int c,int lv,int ans)
10 {
11     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;
12     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);
13 }
14 int main()
15 {
16     ios::sync_with_stdio(0);
17     cin>>n>>m,beg=sqrt(n)+1;
18     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);
19     cout<<(s==INT_MAX?0:s);
20     return 0;
21 }

 

POJ 1190 生日蛋糕题解