P1731 [NOI1999] 生日蛋糕——典型的回溯和剪枝题目，值得一看

本文主要是介绍P1731 [NOI1999] 生日蛋糕——典型的回溯和剪枝题目，值得一看，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

今天尝试了一下md的编辑器，不知道有没有什么改变

[NOI1999] 生日蛋糕

题目背景

数据加强版 link

题目描述

7 月 17 日是 Mr.W 的生日，ACM-THU 为此要制作一个体积为 $N\pi$ 的 $M$ 层生日蛋糕，每层都是一个圆柱体。

设从下往上数第 $i$ （ $\leq i \leq M$ ）层蛋糕是半径为 $R_i$ ，高度为 $H_i$ 的圆柱。当 $\lt M$ 时，要求 $R_i \gt R_{i+1}$ 且 $H_i \gt H_{i+1}$ 。

由于要在蛋糕上抹奶油，为尽可能节约经费，我们希望蛋糕外表面（最下一层的下底面除外）的面积 $Q$ 最小。

请编程对给出的 $N$ 和 $M$ ，找出蛋糕的制作方案（适当的 $R_i$ 和 $H_i$ 的值），使 $S=\dfrac{Q}{\pi}$ 最小。

（除 $Q$ 外，以上所有数据皆为正整数）

输入格式

第一行为一个整数 $N$ （ $\leq 2 \times 10^4$ ），表示待制作的蛋糕的体积为 $N\pi$ 。

第二行为 $M$ （ $\leq 15$ ），表示蛋糕的层数为 $M$ 。

输出格式

输出一个整数 $S$ ，若无解，输出 $0$ 。

样例 #1

样例输入 #1

100
2

样例输出 #1

这个题是一道显然的回溯加剪枝的题目，首先我们要把基础的方法写出来，再来剪枝。所以我们很轻松就可以想到这样的解法：
1.枚举层数（从n到1），半径，高度，体积以及面积，在层数达到0并且体积恰好等于n的时候将面积与面积的最小值进行比较，如果比他小就更改。
2.在函数内部枚举半径i和高度h，范围题目已经告诉你了。
3.当已选中一个高度和一个半径的时候，就按照题目的要求进行下一次的递归。
也就是这样

#include<bits/stdc++.h>
using namespace std;int n,m,minn=INT_MAX;void dfs(int step,int r,int h,int s,int v){if (step==0&&v==n){minn=min(s,minn);return;}for (int i=r-1;i>=step;i--){if (step==m)s=i*i;//这个地方很重要，一定要记得将下表面存着for (int j=h-1;j>=step;j--){dfs(step-1,i,j,s+2*i*j,v+i*i*j);}}return;
}
int main(){cin>>n>>m;dfs(m,n,n,0,0);if (minn==INT_MAX)cout<<0;else cout<<minn;return 0;
}

接下来就是肯定是剪枝了，怎么剪呢？很容易就可以想到如果当前的面积加上往后最小的面积和都要比最小的面积情况大的话，直接退出，同样又有如果当前的v已经大于了n也是直接退出，再有如果当前的v从而算出来的s也是大于最小的面积的话，也是直接退出。代码如下：

#include<bits/stdc++.h>
using namespace std;int n,m,minn=INT_MAX;
int mins[30000],minv[30000];void dfs(int step,int r,int h,int s,int v){if (step==0){if (v==n)minn=min(s,minn);return;}if (s+mins[step-1]>=minn)return;if (v+minv[step-1]>n)return;if(2.0*(n-v)/r+s>=minn)  return;//要用小数，不能用整数，要不然要出错for (int i=r-1;i>=step;i--){if (step==m)s=i*i;for (int j=h-1;j>=step;j--){dfs(step-1,i,j,s+2*i*j,v+i*i*j);}}return;
}
int main(){cin>>n>>m;for (int i=1;i<=m;i++){mins[i]=mins[i-1]+2*i*i;//记录面积的最小minv[i]=i*i*i;//记录体积的最小}dfs(m,n,n,0,0);if (minn==INT_MAX)cout<<0;else cout<<minn;return 0;
}

那么到了这里，基本上函数根本上的优化是没有了，这时候我们就要考虑一下优化循环了：
半径肯定是不能优化了，那么我们就要考虑一下高度了，能不能优化呢？显然是可以的，我们可以用总的体积减去当前的体积和后面最小的体积和再来除以i²得到后面的h并与当前的h进行比较，从而缩短循环的时间，代码如下：

#include<bits/stdc++.h>
using namespace std;int n,m,minn=INT_MAX;
int mins[30000],minv[30000];void dfs(int step,int r,int h,int s,int v){int maxh=h;if (step==0){if (v==n)minn=min(s,minn);return;}if (s+mins[step-1]>=minn)return;if (v+minv[step-1]>n)return;if(2.0*(n-v)/r+s>=minn)  return;for (int i=r-1;i>=step;i--){if (step==m)s=i*i;maxh=min(h-1,(n-minv[step-1]-v)/i/i);for (int j=maxh;j>=step;j--){dfs(step-1,i,j,s+2*i*j,v+i*i*j);}}return;
}
int main(){cin>>n>>m;for (int i=1;i<=m;i++){mins[i]=mins[i-1]+2*i*i;minv[i]=i*i*i;}dfs(m,n,n,0,0);if (minn==INT_MAX)cout<<0;else cout<<minn;return 0;
}