luogu P3648 [APIO2014]序列分割

背景：

逃掉了物理竞赛，有愧于班主任啊。
大概写完这一篇就$blog5$了，暗喜…

题目传送门：

https://www.luogu.org/problemnew/show/P3648

题意：

$n$个数，数的值是$a_i$，分成$m$组，其中每一组编号要连续，某一组（第$k$组）（$[l,r]$）的贡献是$x_k={\sum }_{i=l}^r{a}_i$，总贡献的计算公式是${\sum }_{i=1}^m{x}_i\ast x_{i-1}$.求最大贡献。

思路：

有点像这一题。
考虑常规做法，设$f_{i,l}$表示将前$i$个元素分为$l$组的最大贡献。
设$su{m}_k={\sum }_{i=1}^k{a}_i$。
则状态转移方程为：$f_{i,l}={\max }_{j=1}^{i-1}{f}_{j,l-1}+su{m}_j(su{m}_i-su{m}_j)$.
得到斜率方程：$\frac{(f_{j,l-1}-su{m}_j^2)+(f_{k,l-1}-su{m}_k^2)}{su{m}_j-su{m}_k}<su{m}_i$
然后直接套即可。
然后就做完了。
注意一下当$sum[x]==sum[y]$时的边界条件即可。

据说可以用$WQS$二分（凸优化），先留个坑。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
using namespace std;
	int n,m;
	LL a[100010],sum[100010],f[100010][210];
	int que[100010],pre[100010][210];
LL calc1(int x,int y)
{
	return sum[y]-sum[x];
}
LL calc2(int x,int y,int z)
{
	return (f[x][z-1]-sum[x]*sum[x])-(f[y][z-1]-sum[y]*sum[y]);
}
double calc(int x,int y,int z)
{
	if(sum[x]==sum[y]) return -1e15;
	return (double)(calc2(x,y,z))/(double)(calc1(x,y));
}
int main()
{
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
		sum[i]=sum[i-1]+a[i];
	}
	for(int l=1;l<=m;l++)
	{
		int head=1,tail=1;
		que[1]=0;
		for(int i=1;i<=n;i++)
		{
			while(head<tail&&calc(que[head],que[head+1],l)<=(double)sum[i]) head++;
			f[i][l]=f[que[head]][l-1]+sum[que[head]]*(sum[i]-sum[que[head]]);
			pre[i][l]=que[head];
			while(head<tail&&calc(que[tail],i,l)<=calc(que[tail-1],que[tail],l)) tail--;
			que[++tail]=i;
		}
	}
	printf("%lld\n",f[n][m]);
	int tmp=n;
	for(int i=m;i>=1;i--)
		tmp=pre[tmp][i],printf("%d ",tmp);
}