This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
struct func{
int poz;
int h;
};
double f(func x,int i)
{
double sol=(double)x.h+(double)sqrt(abs(i-x.poz));
return sol;
}
int N;
struct liChao{
vector<func> tr;
void init(func start)
{
tr.resize(4*N);
fill(tr.begin(),tr.end(),start);
}
void set(func x,int i=1,int l=0,int r=N)
{
int m=(l+r)>>1;
bool lft=f(x,l)>f(tr[i],l);
bool mid=f(x,m)>f(tr[i],m);
//printf("%i %i, %i %i %i %i %i %i\n",x.h,x.poz,i,l,r,m,lft,mid);
if(mid){
swap(tr[i],x);
}
if(r-l==1)
return;
if(lft!=mid)
{
set(x,2*i,l,m);
}
else
{
set(x,2*i+1,m,r);
}
}
double get(int poz,int i=1,int l=0,int r=N)
{
int m=(l+r)>>1;
if(r-l==1)
return f(tr[i],poz);
if(poz<m)
return max(f(tr[i],poz),get(poz,2*i,l,m));
else
return max(f(tr[i],poz),get(poz,2*i+1,m,r));
}
} ST,ST2;
int main()
{
int n;
scanf("%i",&n);
N=n;
vector<int> h(n);
for(int i=0;i<n;i++)
{
scanf("%i",&h[i]);
func tr;
tr.h=h[i];
tr.poz=i;
if(i==0)
ST.init(tr);
else
ST.set(tr);
}
for(int i=n-1;i>=0;i--)
{
func tr;
tr.h=h[i];
tr.poz=i;
if(i==n-1)
ST2.init(tr);
else
ST2.set(tr);
}
for(int i=0;i<n;i++)
{
double d=max(ST.get(i),ST2.get(i));
//printf("%i: %f\n",i,d);
int a=d;
//printf("%i\n",a);
if(d>a)
a++;
printf("%i\n",a-h[i]);
}
return 0;
}
Compilation message (stderr)
pio.cpp: In function 'int main()':
pio.cpp:58:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%i",&n);
~~~~~^~~~~~~~~
pio.cpp:63:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%i",&h[i]);
~~~~~^~~~~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |