BZOJ 4518: [Sdoi2016]征途 斜率优化

4518: [Sdoi2016]征途

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 1213  Solved: 668
[Submit][Status][Discuss]

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

Sample Output

36

HINT

1≤n≤3000,保证从 S 到 T 的总路程不超过 30000

斜率优化懒得写题解

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<queue>
#include<set>
#include<map>
using namespace std;
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return f*x;
}
const int N=3010;
int n,m,f[N],a[N],sum[N],head[N],tail[N],q[N];
int dp[N][N];
inline double slop(int x,int y,int k)
{k--;return (dp[x][k]-dp[y][k]+sum[x]*sum[x]-sum[y]*sum[y])/(sum[x]-sum[y]);}
int main()
{
	n=read();m=read();
	for(int i=1;i<=n;i++)a[i]=read();
	for(int i=1;i<=n;i++)sum[i]=sum[i-1]+a[i];
	for(int i=1;i<=n;i++)dp[i][1]=sum[i]*sum[i];
	for(int j=2;j<=m;j++)
	{
		int head=1,tail=1;
		for(int i=1;i<=n;i++)
		{
			while(head<tail&&slop(q[head],q[head+1],j)<2*sum[i])head++;
			int temp=q[head];
			dp[i][j]=dp[temp][j-1]+(sum[i]-sum[temp])*(sum[i]-sum[temp]);
			while(head<tail&&slop(q[tail],i,j)<slop(q[tail-1],q[tail],j))tail--;
			q[++tail]=i;
		}
	}
	int ans=dp[n][m]*m-sum[n]*sum[n];
	printf("%d\n",ans);
	return 0;
}
/*
5 2
1 2 5 8 6

36
*/