饼干

圣诞老人共有M个饼干，准备全部分给N个孩子。

每个孩子有一个贪婪度，第 i 个孩子的贪婪度为 g[i]。

如果有 a[i] 个孩子拿到的饼干数比第 i 个孩子多，那么第 i 个孩子会产生 g[i]*a[i]的怨气。

给定N、M和序列g，圣诞老人请你帮他安排一种分配方式，使得每个孩子至少分到一块饼干，并且所有孩子的怨气总和最小。

输入格式
第一行包含两个整数N,M。

第二行包含N个整数表示g1~gN。

输出格式
第一行一个整数表示最小怨气总和。

第二行N个空格隔开的整数表示每个孩子分到的饼干数，若有多种方案，输出任意一种均可。

数据范围
1≤N≤30,
N≤M≤5000,
1≤gi≤10^7

输入样例：
3 20
1 2 3
输出样例：
2
2 9 9

#include <bits/stdc++.h>

#define pii pair<int,int>
#define fi first
#define se second
using namespace std;
const int N = 35, M = 5e3 + 10;
pii g[N], pre[N][M];
int f[N][M], s[N], n, m, ans[N];

int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++)scanf("%d", &g[i].fi), g[i].se = i;
    sort(g + 1, g + 1 + n, greater<pii >());
    for (int i = 1; i <= n; i++)
        s[i] = s[i - 1] + g[i].fi;
    memset(f, 0x3f, sizeof(f));
    f[0][0] = 0;
    for (int i = 1; i <= n; i++)
        for (int j = i; j <= m; j++) {
            f[i][j] = f[i][j - i];
            pre[i][j] = {i, j - i};
            for (int k = 1; k <= i; k++) {
                int val = f[i - k][j - k] + (s[i] - s[i - k]) * (i - k);
                if (f[i][j] > val)
                    f[i][j] = val, pre[i][j] = {i - k, j - k};
            }
        }
    printf("%d\n", f[n][m]);
    int i = n, j = m, h = 0;
    while (i) {
        if (pre[i][j] == make_pair(i, j - i))
            j -= i, h++;
        else {
            for (int k = i; k > pre[i][j].fi; k--)
                ans[g[k].se] = h + 1;
            pii temp = pre[i][j];
            i = temp.fi, j = temp.se;
        }
    }
    for (int k = 1; k <= n; k++)
        printf("%d ", ans[k]);
    return 0;
}