又一道斜率dp裸题2333。
朴素dp
for(int i=1;i<=n;i++)
read(C[i]);
for(int i=1;i<=n;i++)
sum[i]=sum[i-1]+C[i];
memset(f,63,sizeof(f));
f[1]=(sum[1]-L)*(sum[1]-L);
for(int i=2;i<=n;i++)
{
for(int j=1;j<i;j++)
f[i]=min(f[i],f[j](i-j-1+sum[i]-sum[j]-L)*(i-j-1+sum[i]-sum[j]-L));
}
之后推斜率,这道题还是挺繁琐的。。
(double)(f[i]-f[j]+sum[i]sum[i]-sum[j]*sum[j]+2.0*L(sum[i]-sum[j]))/(double)(2.0*(sum[i]-sum[j]))
长啊。。。
套模板
#include<bits/stdc++.h>
#define N 50000
using namespace std;
int n,L;
int C[N+1];
long long sum[N+1];
long long f[N+1];
int q[N+1],l,r;
template <typename T>
void read(T &x)
{
x=0;
char c=getchar();
while(c>'9'||c<'0')c=getchar();
while(c<='9'&&c>='0')x=x*10+c-'0',c=getchar();
}
double slope(int i,int j)
{
return (double)(f[i]-f[j]+sum[i]*sum[i]-sum[j]*sum[j]+2.0*L*(sum[i]-sum[j]))/(double)(2.0*(sum[i]-sum[j]));
}
int main()
{
read(n),read(L);
for(int i=1;i<=n;i++)
read(C[i]);
for(int i=1;i<=n;i++)
sum[i]=sum[i-1]+C[i];
/*
memset(f,63,sizeof(f));
f[1]=(sum[1]-L)*(sum[1]-L);
for(int i=2;i<=n;i++)
{
for(int j=1;j<i;j++)
f[i]=min(f[i],f[j](i-j-1+sum[i]-sum[j]-L)*(i-j-1+sum[i]-sum[j]-L));
}
cout<<f[n];
*/
/*----------斜率优化--------------*/
L++;
for(int i=1;i<=n;i++)sum[i]+=i;
l=1,r=0,q[++r]=0;
for(int i=1;i<=n;i++)
{
while(l<r&&slope(q[l+1],q[l])<=sum[i])l++;
int tmp=q[l];
f[i]=f[tmp]+(sum[i]-sum[tmp]-L)*(sum[i]-sum[tmp]-L);
while(l<r&&slope(i,q[r])<slope(q[r],q[r-1]))r--;
q[++r]=i;
}
cout<<f[n];
return 0;
}