4518: [Sdoi2016]征途

4518: [Sdoi2016]征途

Time Limit: 10 Sec   Memory Limit: 256 MB
Submit: 784   Solved: 453
[ 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

Source

鸣谢Menci上传

[ Submit][ Status][ Discuss]
先对方差的式子进行化简 其中A[i]:第i段走到路程，，s:总路程 发现其实答案可以分段统计最后再加一个常数 那么令f[i][j]:前i段路程分成j段，最小代价和 假设k后有一位置t，用它转移不比位置k劣，代入上式，化简，得 单调队列维护一个斜率单调递增就行了
注意斜率优化时左式要为定值。。。。调了半天才发现第一次拆的式子左式会改变，显然有误。。。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<vector>
#include<queue>
#include<set>
#include<map>
#include<stack>
#include<bitset>
#include<ext/pb_ds/priority_queue.hpp>
using namespace std;

const int maxn = 3030;
typedef double DB;
typedef long long LL;
const LL INF = 1E16;
const DB eps = 1E-9;

LL s,m,sum[maxn],f[maxn][maxn];
int n,head,tail,Q[maxn];

LL Work(int k,int i,int j)
{
	LL dt = sum[i] - sum[k];
	return f[k][j-1] + m*dt*dt - 2LL*s*dt;
}

DB slope(int k,int t,int j)
{
	LL A = f[t][j-1] - f[k][j-1] + (sum[t] - sum[k])*(2LL*s + m*(sum[t] + sum[k]));
	LL B = m*(sum[t] - sum[k]);
	return (DB)(A) / (DB)(B);
}

bool fcmp(DB x,DB y)
{
	return x > y || fabs(x - y) <= eps;
}

int main()
{
	#ifdef DMC
		freopen("DMC.txt","r",stdin);
	#endif
	
	cin >> n >> m;
	for (int i = 1; i <= n; i++)
		scanf("%lld",&sum[i]),sum[i] += sum[i-1];
	for (int i = 0; i <= n; i++)
		for (int j = 0; j <= m; j++)
			f[i][j] = INF;
	s = sum[n]; f[0][0] = 0;
	for (int j = 1; j <= m; j++)
	{
		Q[head = tail = 1] = j - 1;
		for (int i = j; i <= n; i++)
		{
			while (head < tail && Work(Q[head],i,j) >= Work(Q[head+1],i,j)) ++head;
			f[i][j] = Work(Q[head],i,j);
			if (f[i][j-1] != INF)
			{
				while (head < tail && fcmp(slope(Q[tail-1],Q[tail],j),slope(Q[tail],i,j))) --tail;
				Q[++tail] = i;
			}
		}
	}
	cout << f[n][m] + s*s;
	return 0;
}