• 【CH5105】Cookies


    也是一道线型动态规划的好题……

    读入每个人的贪婪度之后,对其按照从大到小的顺序排序,定义状态f[i][j]为前i个人(排序后)分j个饼干的答案,那么答案为f[n][m],考虑状态转移方程。

    1、若给第i个人的饼干数大于1 ,那么我们将这i个人的饼干数都减1(总共减n),他们的怨气值是不会改变的,因而这种情况下,f[i][j]=f[i][j-i].

    2、若给第i个人的饼干数等于1,那么我们枚举一个k(0≤k<i),表示从k之后一直到i所有的人的饼干数都是1,那么f[i][j]=f[k][j-(i-k)]+k*∑g[c[p]]    (k<p<=i).

    我们先预处理出g数组的前缀和,即可实现O(n)的转移。

    综上,我们在两种决策中取最优即可。另外,本题要求输出方案,我们只需在状态转移时记录每个状态的前驱即可。

     1 #include <iostream>
     2 #include <cstdio>
     3 #include <cstring>
     4 #include <algorithm>
     5 using namespace std;
     6 int n,m,f[40][5010],a[40][5010],b[40][5010];
     7 int s[50],ans[50];
     8 int g[50],c[50];
     9 bool cmp(int x,int y) {
    10     return g[x]>g[y];
    11 }
    12 void calc(int x,int y) {
    13     if(!x) return ;
    14     calc(a[x][y],b[x][y]);
    15     if(a[x][y]==x) 
    16         for(int i=1;i<=x;i++) ans[c[i]]++;
    17     else 
    18         for(int i=a[x][y]+1;i<=x;i++) ans[c[i]]=1;
    19 }
    20 int main() {
    21     scanf("%d%d",&n,&m);
    22     for(int i=1;i<=n;i++)  {
    23         scanf("%d",&g[i]);
    24         c[i]=i;
    25     }
    26     sort(c+1,c+n+1,cmp);
    27     memset(f,0x3f,sizeof(f));
    28     f[0][0]=0;
    29     for(int i=1;i<=n;i++) s[i]=s[i-1]+g[c[i]];
    30     for(int i=1;i<=n;i++)
    31         for(int j=i;j<=m;j++) {
    32             f[i][j]=f[i][j-i];
    33             a[i][j]=i;
    34             b[i][j]=j-i;
    35             for(int k=0;k<i;k++)
    36                 if(f[i][j]>f[k][j-(i-k)]+k*(s[i]-s[k])) {
    37                     f[i][j]=f[k][j-(i-k)]+k*(s[i]-s[k]);
    38                     a[i][j]=k;
    39                     b[i][j]=j-(i-k);
    40                 }
    41         }
    42     printf("%d
    ",f[n][m]);
    43     calc(n,m);
    44     for(int i=1;i<=n;i++) printf("%d ",ans[i]);
    45     return 0;
    46 }
    AC Code
  • 相关阅读:
    cs61b project1
    CS61b lab5
    leetcode DP
    Leetcode 还未解决的bug
    Git使用总结
    Mac TensorFlow Anaconda
    eclipse C++ ld: 1 duplicate symbol for architecture x86_64
    Leetcode Hashtable 1-5
    EC 601 PYTHONPATH
    EC 601 OpenCV Install
  • 原文地址:https://www.cnblogs.com/shl-blog/p/10660961.html
Copyright © 2020-2023  润新知