题目
代码
WQS二分+斜率优化
公式简化看这https://www.luogu.org/blog/xnzy1314/wang-qing-ti-xie-post
其他就很套路了,只要在f转移时加上一个mid。我用c来记录选了多少个(x坐标),所以最后判断c[n]与m的大小关系,如果c[n]小了,则要选多一点,即mid要小一些。
据说还可以用四边形不等式来做,%%%https://www.cnblogs.com/AKCqhzdy/p/9898577.html
代码
#include<cstdio>
#include<cstring>
#include<algorithm>
#define sqr(x) (x)*(x)
using namespace std;
typedef long long ll;
const int MAXN=100010;
const int MAXSIZE=1<<16;
char buf[MAXSIZE],*p1=buf,*p2=buf;
#define gc p1==p2 && (p2=(p1=buf)+fread(buf,1,MAXSIZE,stdin),p1==p2)?EOF:*p1++//debug p1==p1
//#define gc getchar()
inline int read()
{
int re=0;char ch=gc;
while(ch<'0' || ch>'9') ch=gc;
while(ch>='0' && ch<='9') re=re*10+(ch^48),ch=gc;
return re;
}
int n,m;
ll s[MAXN];
ll f[MAXN];int c[MAXN];
inline double X(int k){return s[k];}
inline double Y(int k){return f[k]+sqr(s[k])-2*s[k];}
inline double slope(int i,int j){return (Y(j)-Y(i))/(X(j)-X(i));}
int head,tail,q[MAXN];
int check(ll mid)//debug bool check
{
head=0,tail=0;q[0]=0;
for(int i=1;i<=n;i++)
{
while(head<tail && slope(q[head],q[head+1])<2*s[i]) head++;
int j=q[head];
f[i]=f[j]+sqr(s[i]-s[j]+1)+mid,c[i]=c[j]+1;
while(head<tail && slope(q[tail-1],q[tail])>slope(q[tail],i)) tail--;
q[++tail]=i;
}
return c[n];
}
int main()
{
n=read(),m=read();
for(int i=1;i<=n;i++) s[i]=s[i-1]+(ll)read();
ll l=0,r=1e18,ans;
while(l<=r)
{
ll mid=l+r>>1;//debug int mid=
if(check(mid)<=m) ans=f[n]-mid*m,r=mid-1;
else l=mid+1;
}
printf("%lld\n",ans);
return 0;
}