[SDOI2016]征途 —— 斜率优化DP

时隔多年没有碰斜率优化了。。。
想当年被斜率优化虐的死去活来,现在看看。。。也就那样吧。

Pine开始了从S地到T地的征途。
从S地到T地的路可以划分成n段,相邻两段路的分界点设有休息站。
Pine计划用m天到达T地。除第m天外,每一天晚上Pine都必须在休息站过夜。所以,一段路必须在同一天中走完。
Pine希望每一天走的路长度尽可能相近,所以他希望每一天走的路的长度的方差尽可能小。
帮助Pine求出最小方差是多少。
设方差是v,可以证明,v×m2是一个整数。为了避免精度误差,输出结果时输出v×m2。
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

首先根据方差的定义我们可以将方差的公式\(s^{2}=\frac{\sum_{i=1}^{n}(x_{i}-\bar x)^{2}}{n}\)转换为\(s^{2}=\frac{\sum_{i=1}^{n}x_{i}^2}{m}-\frac{sum_{n}^{2}}{m^{2}}\)
\(sum\)数组为前缀和)
(具体转换过程可以百度百科方差)

对于式子中的\(\frac{sum_{n}^{2}}{m^{2}}\)是已经确定的常数,目前不用考虑。于是我们要求的便是\(min(\sum_{i=1}^{n}x_{i}^2)\)

于是题目就被我们转换为了将\(n\)段路分为\(m\)个块,使它们的平方和最小,不难得出DP的状态转移方程是:

\(f_{i,j}=min(f_{k,j-1}+(sum_{i}-sum_{k})^{2},f_{i,j})\)

只需要枚举k即可,其中状态的第一维定义为第\(i\)段路,第二维定义为第\(j\)个块。

然而不难发现时间复杂度过高,于是可以使用斜率优化。详见代码

#include <bits/stdc++.h>
using namespace std;

#define N 3100
#define LL long long

int n,A[N],S[N],m,q[3*N];
LL f[N][N]; 

int fucky (int a,int j) {
	return f[a][j-1]+S[a]*S[a];
}

int fuck (int a,int b,int j) {
	return (fucky(a,j)-fucky(b,j))/(S[a]-S[b]);
}

int main() {
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>A[i],S[i]=S[i-1]+A[i];	
	for(int i=1;i<=n;i++) f[i][1]=S[i]*S[i];
	for(int j=2;j<=m;j++) {
		int l=1,r=1;
		for(int i=1;i<=n;i++) {
			while(l<r && fuck(q[l],q[l+1],j)<2*S[i]) l++;
			int BestPos=q[l];
			f[i][j]=f[BestPos][j-1]+(S[i]-S[BestPos])*(S[i]-S[BestPos]);
			while(l<r && fucky(q[r],i,j)<fucky(q[r-1],q[r],j)) r--;
			q[++r]=i;
		}
	}
	cout<<f[n][m]*m-S[n]*S[n];
}
posted @ 2019-10-05 17:00  MisakaMKT  阅读(157)  评论(0编辑  收藏  举报