[BZOJ2216][POI2011]Lightning Conductor && 1D1D

题目描述

已知一个长度为n的序列a1,a2,...,an。

对于每个1<=i<=n，找到最小的非负整数p满足 对于任意的j, aj < = ai + p - sqrt(abs(i-j))

输入

第一行n，(1<=n<=500000)

下面每行一个整数，其中第i行是ai。(0<=ai<=1000000000)

输出

n行，第i行表示对于i，得到的p

样例输入

Copy (如果复制到控制台无换行，可以先粘贴到文本编辑器，再复制)

6
5
3
2
4
2
4

样例输出

2
3
5
3
5
4

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<cmath>
#include<queue>
#define SF scanf
#define PF printf
using namespace std;
typedef long long LL;
const int MAXN = 500000;
int a[MAXN+10], n;
double Left[MAXN+10], Right[MAXN+10];
struct Node {
    int pos, l, r;
    Node () {}
    Node (int p, int L, int R) : pos(p), l(L), r(R) {}
} Q[MAXN+10];
double calc(int i, int j) {
    return a[i] + sqrt(abs(i-j));
}
int search(int L, int R, int i, int j) {
    while(L < R) {
        int mid = (L + R) >> 1;
        if(calc(i, mid) < calc(j, mid)) L = mid+1;
        else R = mid;
    }
    return L;
}
void dp(double ans[]) {
    int F =0, R = 0;
    for(int i = 1; i <= n; i++) {
        if(F < R && ++Q[F].l > Q[F].r) F++;
        if(F >= R || calc(i, n) > calc(Q[R-1].pos, n)) {
            while(F < R && calc(i, Q[R-1].l) > calc(Q[R-1].pos, Q[R-1].l)) R--;
            if(F >= R) Q[R++] = Node(i, i, n);
            else {
                int st = search(Q[R-1].l, Q[R-1].r, i, Q[R-1].pos);
                Q[R-1].r = st-1; Q[R++] = Node(i, st, n);
            }
        }
        ans[i] = calc(Q[F].pos, i);
    }
}
int main() {
    SF("%d", &n);
    for(int i = 1; i <= n; i++) SF("%d", &a[i]);
    dp(Left); reverse(a+1, a+1+n); 
    dp(Right); reverse(Right+1, Right+1+n);
    reverse(a+1, a+1+n);
    for(int i = 1; i <= n; i++) PF("%d\n", (int)ceil(max(Left[i], Right[i]) - a[i]));
}