Submission #84883

#	Time	Username	Problem	Language	Result	Execution time	Memory
84883		nikolapesic2802	Lightning Conductor (POI11_pio)	C++14	36 / 100	681 ms	38768 KiB

pio

#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,int p)
{
    if(p==0)
    {
        if(i>x.poz)
            return -1;
    }
    else
        if(i<x.poz)
            return -1;
    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,0)>f(tr[i],l,0);
        bool mid=f(x,m,0)>f(tr[i],m,0);
        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,0);
        if(poz<m)
            return max(f(tr[i],poz,0),get(poz,2*i,l,m));
        else
            return max(f(tr[i],poz,0),get(poz,2*i+1,m,r));
    }
} ST;
struct liChao2{
    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,1)>f(tr[i],l,1);
        bool mid=f(x,m,1)>f(tr[i],m,1);
        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,1);
        if(poz<m)
            return max(f(tr[i],poz,1),get(poz,2*i,l,m));
        else
            return max(f(tr[i],poz,1),get(poz,2*i+1,m,r));
    }
} 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:102:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%i",&n);
     ~~~~~^~~~~~~~~
pio.cpp:107:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%i",&h[i]);
         ~~~~~^~~~~~~~~~~~

• Subtask 1: 0 / 9
• Subtask 2: 9 / 9
• Subtask 3: 9 / 9
• Subtask 4: 9 / 9
• Subtask 5: 9 / 9
• Subtask 6: 0 / 9
• Subtask 7: 0 / 9
• Subtask 8: 0 / 9
• Subtask 9: 0 / 9
• Subtask 10: 0 / 9
• Subtask 11: 0 / 10