POJ-1190 生日蛋糕 NOI99

深搜+几个剪枝。

貌似搜索顺序也挺重要的。。。我不知是不是因为这个然后Tle了好久。。。

#include <cstdio>
#include <iostream>
#include <fstream>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <queue>

#define rep(i, l, r) for(int i=l; i<=r; i++)
#define down(i, l, r) for(int i=l; i>=r; i--)
#define maxn 23
#define MAX 1<<30

using namespace std;

int n, m, ans, minv[maxn], mins[maxn];

void Search(int v, int s, int l, int R, int H)
{
    if (!l)
    {
        if (v == n && s < ans)
            ans = s;
        return;
    }
    if (s+mins[l-] >= ans || v+minv[l-] > n || (n-v)* >= R*(ans-s)) return;
    down(r, R-, l)
    {
        if (l == m) s = r*r;
        int tmp = min((n-v-minv[l-])/(r*r), H-);
        down(h, tmp, l) Search(v+r*r*h, s+*r*h, l-, r, h);
    }
}

int main()
{
    scanf("%d%d", &n, &m); ans = n+;
    rep(i, , m) minv[i] = minv[i-]+i*i*i;
    rep(i, , m) mins[i] = mins[i-]+*i*i;
    Search(, , m, sqrt((double)(n-minv[m-])/m)+, (n-minv[m-])/(m*m)+);
    if (ans == n+) ans = ; printf("%d\n", ans);
    return ;
}