[NOI1999] 生日蛋糕(DFS之剪枝与优化)

洛谷

acwing 推导

蛋糕的体积为n，有m层，使得表面积最小（最下一层的下底面除外）
当 i<M时，要求 R i >R i+1且 H i >H i+1。

#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>

using namespace std;

const int N = 25, INF = 1e9;

int n, m;
int minv[N], mins[N];//前面若干层的体积和侧面积最小值 
int R[N], H[N];//每层的 
int ans = INF;

void dfs(int u, int v, int s)//从底层向上枚举，下面层的最小体积和面积
{
    if (v + minv[u] > n) return;//当前体积加上前面u层的体积大于n
    if (s + mins[u] >= ans) return;
    if (s + 2 * (n - v) / R[u + 1] >= ans) return;
    //（u层下面的最小面积）s+估算的u层上面面积（包括u层）如果大于ans
    //n-v =R[u] ^ 2 * H[u]   s[u] = H[u] * R[u] > H[u] * R[u] ^ 2 /R[u+1]
    if (!u)
    {
        if (v == n) ans = s;
        return;
    }

    for (int r = min(R[u + 1] - 1, (int)sqrt(n - v)); r >= u; r -- )
    //先枚举半径后枚举高度使得总分支数最少，r ——> ^2级别
        for (int h = min(H[u + 1] - 1, (n - v) / r / r); h >= u; h -- )
        {
            int t = 0;
            if (u == m) t = r * r;
            //如果是最底层，加上红色的面积，全部红色面积加起来就是最底层的上表面积
            R[u] = r, H[u] = h;
            dfs(u - 1, v + r * r * h, s + 2 * r * h + t);
        }
}

int main()
{
    cin >> n >> m;

    for (int i = 1; i <= m; i ++ )
    {
        minv[i] = minv[i - 1] + i * i * i;
        mins[i] = mins[i - 1] + 2 * i * i;
    }

    R[m + 1] = H[m + 1] = INF;

    dfs(m, 0, 0);

    if (ans == INF) ans = 0;
    cout << ans << endl;

    return 0;
}