题目大意:给出序列a,对于每一个i,求出max{a[j]+sqrt(abs(i-j))}-a[i](n<=500000)
决策单调性,对于j< k< i,随着i的右移,sqrt(i-j)的增长速度小于sqrt(i-k)。先求出i左边的最优决策点,再求出i右边的最优决策点,就不用讨论绝对值了。
为什么WA了呢?因为题目要求的是向上取整,而比较两个决策点优劣的时候必须用没取整之前的double比,否则两个不同的值可能取完整就相等了。
自己的代码:
#include<cstdio>
#include<cmath>
#define gm 500001
using namespace std;
int f[gm];
int stk[gm],top=0;
int left[gm],right[gm];
int n;
int a[gm];
inline double va(int i,int j) {return a[j]+sqrt(i-j);}
inline double vl(int i,int j) {return a[j]+sqrt(j-i);}
inline int max(int a,int b){return a<b?b:a;}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
stk[0]=1;left[0]=1;right[0]=n;
for(int i=2;i<=n;i++)
{
if(va(n,i)<=va(n,stk[top])) continue;
while(~top&&left[top]>=i&&va(left[top],stk[top])<=va(left[top],i)) --top;
stk[++top]=i;
if(!top) {left[0]=i;right[0]=n;continue;}
int l=left[top-1]+1,r=n,mid;
if(l<i) l=i;
while(l<r)
{
mid=(l+r)>>1;
if(va(mid,i)<=va(mid,stk[top-1])) l=mid+1;
else r=mid;
}
left[top]=l;right[top]=n;
right[top-1]=l-1;
}
for(int i=0;i<=top;i++)
for(int j=left[i];j<=right[i];j++)
f[j]=ceil(va(j,stk[i]));
top=0;stk[0]=n;left[0]=1;right[0]=n;
for(int i=n-1;i;i--)
{
if(vl(1,i)<=vl(1,stk[top])) continue;
while(~top&&right[top]<=i&&vl(right[top],stk[top])<=vl(right[top],i)) --top;
stk[++top]=i;
if(!top) {left[0]=1;right[0]=i;continue;}
int l=1,r=right[top-1]-1,mid;
if(i<r) r=i;
while(l<r)
{
mid=(l+r+1)>>1;
if(vl(mid,i)<=vl(mid,stk[top-1])) r=mid-1;
else l=mid;
}
left[top]=1;right[top]=r;
left[top-1]=r+1;
}
for(int i=0;i<=top;i++)
for(int j=left[i];j<=right[i];j++)
f[j]=max(f[j],ceil(vl(j,stk[i])));
for(int i=1;i<=n;i++)
printf("%d\n",f[i]-a[i]);
return 0;
}