bzoj 4850 [Jsoi2016]燈塔

Description
JSOI的國境線上有N一座連續的山峯,其中第ii座的高度是hi??.爲了簡單起見,我們認爲這N座山峯排成了連續一條
直線.如果在第ii座山峯上建立一座高度爲p(p≥0)的燈塔,JYY發現,這座燈塔能夠照亮第jj座山峯,當且僅當滿足如
下不等式:hj≤hi+p-sqrt(|i-j|)JSOI國王希望對於每一座山峯,JYY都能提供建造一座能夠照亮全部其他山峯的燈
塔所需要的最小高度.你能幫助JYY麼?
1< N ≤ 10^5
0 < hi ≤ 10^9
Input
輸入一行包含一個正整數N。
接下來N行，第i行包含一個正整數ℎi，表示第i座山峯的高度。
Output
第i行包含一個非負整數，表示在第i座山峯上修建燈塔所需要的最小高度Pi
Sample Input
6

5

3

2

4

2

4
Sample Output
2

3

5

3

5

4
HINT
Source
https://blog.csdn.net/elijahqi/article/details/80899228

#include<bits/stdc++.h>
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(!isdigit(ch)) {if(ch=='-') f=-1;ch=gc();}
    while(isdigit(ch)) x=x*10+ch-'0',ch=gc();
    return x*f;
}
const int N=500050;
struct node{
    int p,l,r;
}data[N],q[N];
double dp1[N],dp2[N];int n,a[N];
inline double calc(int j,int i){
    return a[j]+sqrt(abs(i-j))-a[i];
}
inline int find(const node &a,int x){
    int l=a.l,r=a.r,p=a.p;
    while(l<=r){
        int mid=l+r>>1;
        if(calc(p,mid)>calc(x,mid)) l=mid+1;else r=mid-1;
    }return l;
}
inline void gao(double *dp){
    int h=1,t=0;
    for (int i=1;i<=n;++i){
        ++q[h].l;if(h<=t&&q[h].l>q[h].r) ++h;
        if(h>t||calc(q[t].p,n)<calc(i,n)){
            while(h<=t&&calc(q[t].p,q[t].l)<calc(i,q[t].l)) --t;
            if (h>t) q[++t]=(node){i,i,n};else{
                int md=find(q[t],i);
                q[t].r=md-1;q[++t]=(node){i,md,n};
            }
        }dp[i]=calc(q[h].p,i);
    }
}
int main(){
//  freopen("bzoj2216.in","r",stdin);
    n=read();
    for (int i=1;i<=n;++i) a[i]=read();gao(dp1);
    for (int i=1,j=n;i<j;++i,--j) swap(a[i],a[j]);gao(dp2);
    for (int i=1;i<=n;++i) printf("%d\n",max(0,(int)ceil(max(dp1[i],dp2[n-i+1]))));
    return 0;
}