Description
Pine开始了从S地到T地的征途。
从S地到T地的路可以划分成n段,相邻两段路的分界点设有休息站。
Pine计划用m天到达T地。除第m天外,每一天晚上Pine都必须在休息站过夜。所以,一段路必须在同一天中走完。
Pine希望每一天走的路长度尽可能相近,所以他希望每一天走的路的长度的方差尽可能小。
帮助Pine求出最小方差是多少。
设方差是v,可以证明,v×m^2是一个整数。为了避免精度误差,输出结果时输出v×m^2。
Input
第一行两个数 n、m。
第二行 n 个数,表示 n 段路的长度
Output
一个数,最小方差乘以 m^2 后的值
Sample Input
5 2
1 2 5 8 6
1 2 5 8 6
Sample Output
36
HINT
1≤n≤3000,保证从 S 到 T 的总路程不超过 30000
方差:s^2=Σ(Ai-x)^2/m
全都乘一个m^2,然后就是斜率优化丝帛题,设f[i][j]表示前j天,现在到了第i段路的最小方差。
S[i]表示前缀和*m,x表示所有数之和。
f[i][j]=min{f[i`][j-1]+(S[i]-S[i`]-x)^2}
倒腾一下式子
f[i][j]=(S[i]-x)^2+ min{(f[i`][j-1]+S[i`]^2)-(2*S[i`])*(S[i]-x)}
y x k
可以看成一条直线在y轴上截距,维护一个下凸壳即可。
#include<cstdio>
#include<cctype>
#include<queue>
#include<cstring>
#include<algorithm>
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define dwn(i,s,t) for(int i=s;i>=t;i--)
#define ren for(int i=first[x];i;i=next[i])
using namespace std;
const int BufferSize=1<<16;
char buffer[BufferSize],*head,*tail;
inline char Getchar() {
if(head==tail) {
int l=fread(buffer,1,BufferSize,stdin);
tail=(head=buffer)+l;
}
return *head++;
}
inline int read() {
int x=0,f=1;char c=Getchar();
for(;!isdigit(c);c=Getchar()) if(c=='-') f=-1;
for(;isdigit(c);c=Getchar()) x=x*10+c-'0';
return x*f;
}
typedef long long ll;
const int maxn=3010;
const ll inf=1ll<<60;
ll f[2][maxn],S[maxn];
int n,m,x,cur,Q[maxn];
double slop(int x,int y) {return (double)(f[cur^1][x]+S[x]*S[x]-f[cur^1][y]-S[y]*S[y])/(2*S[x]-2*S[y]);}
ll getdp(int k,int i) {return f[cur^1][k]+S[k]*S[k]-2*S[k]*(S[i]-x);}
int main() {
n=read();m=read();
rep(i,1,n) S[i]=S[i-1]+read()*m;
x=S[n]/m;
rep(i,1,n) f[0][i]=inf;
rep(j,1,m) {
cur^=1;int l=1,r=0;
rep(i,j,n) {
while(l<r&&slop(Q[r-1],Q[r])>=slop(Q[r],i-1)) r--;Q[++r]=i-1;
while(l<r&&getdp(Q[l],i)>=getdp(Q[l+1],i)) l++;
f[cur][i]=getdp(Q[l],i)+(S[i]-x)*(S[i]-x);
}
}
printf("%lld\n",f[cur][n]/m);
return 0;
}