题意:已知一个长度为n的序列a1,a2,…,an。
对于每个1<=i<=n,找到最小的非负整数p满足 对于任意的j, aj < = ai + p - sqrt(abs(i-j))
p=max(aj+sqrt(abs(i-j))+a[i],首先去掉绝对值符号,从后往前和从前往后各扫一遍,观察式子发现具有决策单调性(或者打表可知?)
对于决策点P,借助单调队列维护点P最优的区间,如果已经超过队首最优的区间则删除队首,加入决策时如果这个决策直到最后还不优则不加入,否则和队尾元素判断,如果能覆盖队尾的区间则删去队尾,不能就求交点(队尾元素的最优区间一定是X-N),想清楚还是很好处理的~
Tips:我本地开O2测是AC,交到OJ上就莫名WA…无语了
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
const int maxn=500000+10;
struct node
{
int p,l,r;
}q[maxn];
int n,head,tail,a[maxn],i;
double f[maxn],g[maxn];
double calc(int i,node x)
{
double res=a[x.p]-a[i]+sqrt(abs(x.p-i));
return res;
}
int work(node a,node b)
{
int l=a.l,r=a.r,ans=0;
while(l<=r)
{
int mid=(l+r)>>1;
if(calc(mid,b)>=calc(mid,a)) r=mid-1,ans=mid;
else l=mid+1;
}
if(!ans) return n+1;
return ans;
}
void dp(double *f)
{
node x;x.p=1;x.l=1,x.r=n;
q[head=tail=1]=x;
for(int i=2;i<=n;i++)
{
q[head].l++;
while(head<tail&&q[head].l>q[head].r) head++;
while(head<tail&&calc(i,q[head])<calc(i,q[head+1])) head++;
f[i]=calc(i,q[head]);
x.p=i;x.l=i,x.r=n;
if(calc(n,q[tail])>calc(n,x)) continue;
while(head<=tail&&calc(q[tail].l,q[tail])<calc(q[tail].l,x)) tail--;
if(head<=tail)
{
int yy=work(q[tail],x);
if(yy<=n)
{
x.l=yy;
q[tail].r=yy-1;
q[++tail]=x;
}
}
else q[++tail]=x;
}
}
int main()
{
freopen("pio.in","r",stdin);
freopen("pio.out","w",stdout);
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
dp(f);
for(int i=1;i<=n/2;i++)
swap(a[i],a[n+1-i]);
dp(g);
for(int i=1;i<=n;i++)
{
double aa=max(f[i],g[n+1-i]);
if((int)aa<aa) aa=(int)aa+1;
printf("%.0lf\n",aa);
}
return 0;
}