题目传送门
有点水啊这道斜率优化。。
解法:
看到亮哥和Rose都做了这道题。
也忍不住去看了看。
看了之后发现挺水的。
推了个特别长的方程。
很显然应该把数组倒过来然后就变成左边的防御塔咯。
然后f[i]表示前i个的花费。
然后f[i]=min(f[j-1]+(i-j)(i-j+1)/2+a[j])。
然后假设k是在j后面,且相对于i来说k更优。
那么对于i后面的,k也肯定更优。
那么就有方程f[k-1]+(i-k)(i-k+1)/2+a[k]<f[j-1]+(i-j)(i-j+1)/2+a[j]
2f[k-1]+(i-k)(i-k+1)+2a[k] <2f[j-1]+(i-j)(i-j+1)+2a[j]
展开
2f[k-1]-ki-ki+k^2-k<2f[j-1]-ji-ji+j^2-j
移过来变成
2f[k-1]-2f[j-1]+k^2-k-j^2+j<2(k-j)i
除过去
(2f[k-1]-2f[j-1]+k^2-k-j^2+j)/(2(k-j))<i
然后就单调队列维护就行了呀。
代码实现:
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
ll f[1100000];
ll list[1100000],head,tail,a[1100000];
ll slop(ll j,ll k) {return (2*(f[k-1]-f[j-1])+k*k-k-j*j+j+2*(a[k]-a[j]))/(2*(k-j));}
int main() {
int n;scanf("%d",&n);
for(int i=1;i<=n;i++)scanf("%lld",&a[n-i+1]);
f[1]=a[1];list[1]=1;head=1;tail=1;
for(int i=2;i<=n;i++) {
while(slop(list[head],list[head+1])<i&&head<tail)head++;
ll l=list[head];f[i]=f[l-1]+(i-l)*(i-l+1)/2+a[l];
while(slop(list[tail],i)<slop(list[tail-1],list[tail])&&head<tail)tail--;
list[++tail]=i;
}
printf("%lld\n",f[n]);
return 0;
}