Gym100340 线性dp

//看题解写的 https://blog.csdn.net/sdfzyhx/article/details/51804748
#include<bits/stdc++.h>
using namespace std;
#define ll long long 

struct node{
    int id,g;
    bool operator<(const node a)const{
        return g>a.g;
    }
}g[];
int n,m,b[][],pre[][];
ll sum[],dp[][];

int main(){
    freopen("cookies.in","r",stdin);
    freopen("cookies.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++)scanf("%d",&g[i].g),g[i].id=i;
    sort(g+,g++n);//降序排列
    for(int i=;i<=n;i++)sum[i]=sum[i-]+g[i].g;
    memset(dp,0x3f,sizeof dp);
    dp[][]=;
    for(int i=;i<=n;i++)
        for(int j=i;j<=m;j++){
            dp[i][j]=dp[i][j-i];
            for(int k=;k<=i;k++){
                if(dp[i][j]>dp[i-k][j-k]+(sum[i]-sum[i-k])*(i-k)){//是保留原状态更优还是k-i区间都发一块饼干的状态更优
                    dp[i][j]=dp[i-k][j-k]+(sum[i]-sum[i-k])*(i-k);
                    b[i][j]=;//这个状态是发了一块饼干的
                    pre[i][j]=i-k;//i-k+1 到 i都只发了一块饼干
                }
            }
        }
    printf("%lld\n",dp[n][m]);
    int p=n,t=m,ans[]={};
    while(p){
        if(b[p][t]){//只发了一块饼干的状态
            int x=pre[p][t];
            for(int i=x+;i<=p;i++)
                 ans[g[i].id]++;
            t-=p-x;p=x;
        }
        else {//第一种状态转移，1-p所有阶梯都下降1
            for(int i=;i<=p;i++)ans[g[i].id]++;
            t-=p;
        }
    }
    for(int i=;i<=n;i++) printf("%d ",ans[i]);
}