NOIP提高组 闭门造车

Description

自从htn体验了一把飙车的快感，他就下定决心要闭门造车！但是他两手空空怎么造得出车来呢?无奈的他只好来到了汽车零部件商店。一走进商店，玲琅满目的各式零件看得htn眼花缭乱。但是他很快便反应过来:我只要买一套好的零件就行。首先它们的性能差不能太大，否则汽车的兼容性不好，开着开着就损坏了；其次，当然是越便宜越好了！为了打造一辆顶级跑车，htn陷入了沉思……

现在商店中有 N 件零件，给出这 N 件零件的价格，其性能等于价格。htn要从中购买一套零件，即选取这个序列的一个子串（连续一段）。要求如下：

1、这一套零件个数要大于等于2（这才算一套）。

2、这套零件的性能差为首尾两个零件的性能差（htn觉得每一个都比较性能差实在是太累了）。

3、购买这套零件的价格和为它们各自价格的总和。

4、最终的总花费为 性能差²+价格和²。

5、由于商店最近有优惠活动，所以每一套零件的第一个都是免费的。对此毫无经验的htn只好向经验丰富的你求助了。

Data Constraint

对于 30% 的数据， N≤1000。

对于 50% 的数据， N≤10000。

对于 100% 的数据， N≤100000。

Solution

我们考虑用单调队列解决，对于所有的价格a，设前i个a的前缀和为s[i]。对当前的i为a[i],假设有d$_1$、d$_2$、d$_3$……、d$_k$这些数,且d$_1$、d$_2$、d$_3$……、d$_k$这些数的值是单调递增的，在a中的位置也是单调递增的。显然，对于一个d[j]、d[k],( j>k，a[i]>d[j]>d[k])，那么显然j~i的贡献$(d[j]-a[i])^2+(d[j]在a中的位置-i)^2$比k~i的贡献$(d[k]-a[i])^2+(d[k]在a中的位置-i)^2$小(因为d数组中的数在a中的位置也是单调递增的)。

所以，对于当前的一个a[i],我们只要计算d中d[j]>a[i]的情况对以i为右端点，d[j]在a中的位置为左端点的更新和计算以d[j]在a中的位置为右端点，d[j-1]在a中的位置为左端点的更新，并将它们从队列中删除，最后将a[i]加入d数组。同理，再同时维护一个单调递减的数组就好。因为只维护一个单调递增的数列会存在偏差。由于每个数只会进出队列一次，所以整个算法的时间复杂度为O(N)。

代码

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn=100005;
ll f[maxn],d[maxn],g[maxn],s[maxn],n,i,t,j,k,l,x,num,a[maxn],ans,num1;
ll sqr(ll x){
    return x*x;
}
int main(){
//  freopen("data.in","r",stdin);
    scanf("%d",&n);
    for (i=1;i<=n;i++)
        scanf("%lld",&a[i]),s[i]=s[i-1]+a[i];
    memset(f,127,sizeof(f));ans=f[1];
    for (i=1;i<=n;i++){
        while (a[d[num]]>a[i]){
            if (num>1) f[d[num]]=min(f[d[num]],sqr(s[d[num-1]]-s[d[num]])+sqr(a[d[num]]-a[d[num-1]]));
            f[i]=min(f[i],sqr(a[i]-a[d[num]])+sqr(s[i]-s[d[num]]));
            num--;
        }
        d[++num]=i;
        while (a[g[num1]]<a[i] && num1>0){
            if (num1>1) f[g[num1]]=min(f[g[num1]],sqr(s[g[num1-1]]-s[g[num1]])+sqr(a[g[num1]]-a[g[num1-1]]));
            f[i]=min(f[i],sqr(a[i]-a[g[num1]])+sqr(s[i]-s[g[num1]]));
            num1--;
        }
        g[++num1]=i;
    }
    for (i=2;i<=num;i++)
        f[d[i]]=min(f[d[i]],sqr(a[d[i]]-a[d[i-1]])+sqr(s[d[i]]-s[d[i-1]]));
    for (i=2;i<=num1;i++)
        f[g[i]]=min(f[g[i]],sqr(a[g[i]]-a[g[i-1]])+sqr(s[g[i]]-s[g[i-1]]));
    for (i=1;i<=n;i++)
        ans=min(ans,f[i]);
    printf("%lld\n",ans);
}