Acwing-277-饼干(DP)

链接:

https://www.acwing.com/problem/content/279/

题意:

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

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

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

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

思路:

先考虑贪心,怨气较大的人得到的饼干肯定较多.
可以先进行排序, 再去进行DP, 考虑Dp使没有确定的值.
对于Dp[i,j]表示前i个孩子,共分配j个饼干的情况.
对于第i个孩子, 考虑将其分配大于1个饼干,可以发现将前i个人的饼干数都减一,并不影响答案的数值.
在考虑从i开始往前k个孩子全部分配1个饼干的情况,挨个查找.同时前缀和记录怨气的总值.
而为了找到每个孩子的具体数值.可以用a,b两个数组记录某个状态是由那个状态转移而来.
结束DP之后.递归查找.当某个状态是由孩子个数相同情况推出来的时候.可以将这些孩子的总数诶个加1.
否则可以得到k+1到n这些孩子的饼干数都为1.
其中还用到了神奇的排序...

代码:

#include <bits/stdc++.h>
using namespace std;
const int INF = 1e9;

int Dp[50][5010];
int G[50];
int Sum[50];
int C[50];
int a[50][5010], b[50][5010];
int Ans[50];
int n, m;

bool cmp(int l, int r)
{
    return G[l] > G[r];
}

void Ser(int n, int m)
{
    if (n == 0)
        return;
    Ser(a[n][m], b[n][m]);
    if (a[n][m] == n)
    {
        for (int i = 1;i <= n;i++)
            Ans[C[i]]++;
    }
    else
    {
        for (int i = a[n][m]+1;i <= n;i++)
            Ans[C[i]] = 1;
    }
}

int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 1;i <= n;i++)
        scanf("%d", &G[i]), C[i] = i;
    for (int i = 0;i <= n;i++)
    {
        for (int j = 0;j <= m;j++)
            Dp[i][j] = INF;
    }
    sort(C+1, C+1+n, cmp);
    Dp[0][0] = 0;
    for (int i = 1;i <= n;i++)
    {
        Sum[i] = Sum[i-1]+G[C[i]];
        for (int j = i;j <= m;j++)
        {
            Dp[i][j] = Dp[i][j-i];
            a[i][j] = i;
            b[i][j] = j-i;
            for (int k = 0;k < i;k++)
            {
                if (Dp[i][j] > Dp[k][j-(i-k)]+k*(Sum[i]-Sum[k]))
                {
                    Dp[i][j] = Dp[k][j-(i-k)]+k*(Sum[i]-Sum[k]);
                    a[i][j] = k;
                    b[i][j] = j-(i-k);
                }
            }
        }
    }
    printf("%d
", Dp[n][m]);
    Ser(n, m);
    for (int i = 1;i <= n;i++)
        printf("%d ", Ans[i]);
    printf("
");

    return 0;
}