code1319 玩具装箱

#include<iostream>
#include<cstring>
#define Size 50005
using namespace std;

int n;
int c[Size];
long long sum[Size];
long long L;
long long dp[Size];
int choice[Size];

long long cost(int l,int r){
    long long x=sum[r]-sum[l-1] + r-l;
    return (x-L)*(x-L);
}

int main(){
    freopen("1319.in","r",stdin); 
    memset(dp,0x7f,sizeof(dp));
    
    cin>>n>>L;
    for(int i=1;i<=n;i++){
        cin>>c[i];
        sum[i]=sum[i-1]+c[i]; 
    }
    
    dp[0]=0;
    choice[0]=0; 
    for(int i=1;i<=n;i++){
        for(int u=choice[i-1];u<=i-1;u++){
            if(dp[u]+cost(u+1,i)<=dp[i]){
                dp[i]=dp[u]+cost(u+1,i);
                choice[i]=u;
            }
            
        }
    }
    
    cout<<dp[n]; 
    
    fclose(stdin); 
    return 0;
}