我最近总是在省选题中找noip题做…
题意:给你一排n个数,让你分成m组,每组是连续的一段,求每段和的最小方差,乘以m^2后输出。n≤3000。
在学斜率优化的时候我记得某位大佬提到了方差,所以就开始画柿子吧,设sum为n个数的和,
xi
为第i段的和。
ans=m∑i=1m(xi−summ)2
然后…
ans=m∑i=1mx2i−sum2
变成了最小化这玩意 ∑mi=1x2i ,就是个裸斜率优化了。
#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>
using namespace std;
#define mmst(a, b) memset(a, b, sizeof(a))
#define mmcp(a, b) memcpy(a, b, sizeof(b))
typedef long long LL;
const int N=3030;
int n,m;
int g[N],a[N],sum[N],q[N],y[N],f[N];
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]),sum[i]=sum[i-1]+a[i];
for(int i=1;i<=n;i++)
g[i]=sum[i]*sum[i],y[i]=g[i]*2;
for(int j=2;j<=m;j++)
{
int hh=1,tt=1;
q[1]=j-1;
for(int i=j;i<=n;i++)
{
while(hh<tt&&y[q[hh+1]]-y[q[hh]]<=2*sum[i]*(sum[q[hh+1]]-sum[q[hh]]))
hh++;
int hy=q[hh];
f[i]=g[hy]+(sum[i]-sum[hy])*(sum[i]-sum[hy]);
y[i]=g[i]+sum[i]*sum[i];
while(hh<tt&&(y[i]-y[q[tt]])*(sum[q[tt]]-sum[q[tt-1]])<=(y[q[tt]]-y[q[tt-1]])*(sum[i]-sum[q[tt]]))
tt--;
q[++tt]=i;
}
for(int i=1;i<=n;i++)
g[i]=f[i];
}
cout<<m*g[n]-sum[n]*sum[n]<<endl;
return 0;
}
在我这个oi蒟蒻眼里,这只是一个发生在身边,却从未涉足的浪漫小故事。