#include "shortcut.h"
#include <bits/stdc++.h>
using namespace std;
const long long INF=1e18;
const int mxn=300001;
struct T{
long long x=-INF,y=INF;
int l=0,r=0;
}st[mxn*20];
long long s[mxn],d[mxn];
int id,idx[mxn],root[mxn];
vector <long long> v;
T operator +(T a, T b){
return {max(a.x,b.x),min(a.y,b.y)};
}
int update(int node, int l, int r, int i){
id++;
if (l==r){
st[id]={s[l]+d[l],s[l]-d[l]};
return id;
}
int cur=id,mid=(l+r)>>1;
st[cur]=st[node];
if (mid<i)
st[cur].r=update(st[node].r,mid+1,r,i);
else
st[cur].l=update(st[node].l,l,mid,i);
st[cur].x=max(st[st[cur].l].x,st[st[cur].r].x);
st[cur].y=min(st[st[cur].l].y,st[st[cur].r].y);
return cur;
}
T get(int node, int l, int r, int u, int v){
if (l>v||r<u||l>r||u>v)
return {-INF,INF};
if (l>=u&&r<=v)
return st[node];
int mid=(l+r)>>1;
return get(st[node].l,l,mid,u,v)+get(st[node].r,mid+1,r,u,v);
}
long long find_shortcut(int n, vector <int> l, vector <int> D, int c){
long long lo=0,hi=INF,kq=-1;
for (int i=0;i<n;i++)
d[i]=D[i];
for (int i=1;i<n;i++){
s[i]=s[i-1]+l[i-1];
v.push_back(s[i]);
}
sort(v.begin(),v.end());
iota(idx,idx+n,0);
sort(idx,idx+n,[](int i, int j){return s[i]+d[i]>s[j]+d[j];});
for (int i=0;i<n;i++)
root[i]=update((i?root[i-1]:0),0,n-1,idx[i]);
while (lo<=hi){
long long mid=(lo+hi)>>1,ch=0,mnx=-INF,mxx=INF,mny=-INF,mxy=INF;
for (int i=0;i<n;i++){
int l=0,r=n-1,pos=-1;
while (l<=r){
int m=(l+r)>>1;
if (s[idx[m]]+d[idx[m]]>mid+s[i]-d[i]){
pos=m;
l=m+1;
}
else
r=m-1;
}
T t=get(root[pos],0,n-1,i+1,n-1);
mnx=max(mnx,s[i]+d[i]+c-mid+t.x);
mxx=min(mxx,s[i]-d[i]-c+mid+t.y);
mny=max(mny,s[i]+d[i]+c-mid-t.y);
mxy=min(mxy,s[i]-d[i]-c+mid-t.x);
}
for (int i=0;i<n;i++){
long long l=max(mnx-s[i],s[i]-mxy),r=min(mxx-s[i],s[i]-mny);
int x=lower_bound(v.begin(),v.end(),l)-v.begin(),y=upper_bound(v.begin(),v.end(),r)-v.begin()-1;
if (x<=y)
ch=1;
}
if (ch){
kq=mid;
hi=mid-1;
}
else
lo=mid+1;
}
return kq;
}
Compilation message
Compilation timeout while compiling shortcut
