Submission #641156

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
641156	2022-09-16T06:03:24 Z	Tenis0206	Santa Claus (RMI19_santa)	C++11	100 / 100	342 ms	10932 KB

santa

#include <bits/stdc++.h>

using namespace std;

int n;

int x[100005],v[100005];
bool h[100005];

int rez[100005],r[100005];

int nr_left = 0;

int ai[500005],lazy[500005];

void preset(int nod, int a, int b)
{
    ai[nod] = lazy[nod] = 0;
    if(a==b)
    {
        return;
    }
    int mij = (a + b) >> 1;
    preset(nod*2,a,mij);
    preset(nod*2+1,mij+1,b);
}

void propag(int nod)
{
    if(!lazy[nod])
    {
        return;
    }
    ai[nod*2] += lazy[nod];
    ai[nod*2+1] += lazy[nod];
    lazy[nod*2] += lazy[nod];
    lazy[nod*2+1] += lazy[nod];
    lazy[nod] = 0;
    return;
}

void update(int ua, int ub, int val, int nod, int a, int b)
{
    if(ua<=a && ub>=b)
    {
        ai[nod] += val;
        lazy[nod] += val;
        return;
    }
    propag(nod);
    int mij = (a + b) >> 1;
    if(ua<=mij)
    {
        update(ua,ub,val,nod*2,a,mij);
    }
    if(ub>mij)
    {
        update(ua,ub,val,nod*2+1,mij+1,b);
    }
    ai[nod] = min(ai[nod*2],ai[nod*2+1]);
}

void add_elf(int val)
{
    update(val,n,-1,1,1,n);
}

void remove_elf(int val)
{
    update(val,n,+1,1,1,n);
}

void add_child(int val)
{
    update(val,n,+1,1,1,n);
}

void remove_child(int val)
{
    update(val,n,-1,1,1,n);
}

bool is_ok()
{
    return (ai[1] >= 0);
}

void solve_test()
{
    cin>>n;
    for(int i=1; i<=n; i++)
    {
        cin>>x[i];
    }
    int last_elf = 0;
    for(int i=1; i<=n; i++)
    {
        cin>>h[i];
        if(!h[i])
        {
            last_elf = i;
        }
    }
    for(int i=1; i<=n; i++)
    {
        cin>>v[i];
    }
    int j = 0;
    multiset<int> s;
    int l_poz = 0;
    for(int i=1;i<=n;i++)
    {
        while(j<n && (!is_ok() || j<last_elf || j<i))
        {
            ++j;
            if(h[j])
            {
                add_child(v[j]);
            }
            else
            {
                add_elf(v[j]);
            }
        }
        if(is_ok())
        {
            r[i] = j;
            l_poz = i;
        }
        else
        {
            break;
        }
        if(h[i]==0)
        {
            s.insert(v[i]);
        }
        else
        {
            remove_child(v[i]);
            auto dr = s.lower_bound(v[i]);
            if(dr!=s.end())
            {
                remove_elf(*dr);
                s.erase(dr);
            }
        }
    }
    r[l_poz+1] = n + 1;
    for(int i=1;i<r[1];i++)
    {
        rez[i] = -1;
    }
    for(int i=1;i<=l_poz;i++)
    {
        for(int j=r[i];j<r[i+1];j++)
        {
            rez[j] = 2 * x[j] - x[i];
        }
    }
    for(int i=1;i<=n;i++)
    {
        cout<<rez[i]<<' ';
    }
    cout<<'\n';
    preset(1,1,n);
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
  //  freopen("nr.in","r",stdin);
  //  freopen("nr.out","w",stdout);
    int t;
    cin>>t;
    for(int test=1; test<=t; test++)
    {
        solve_test();
    }
    return 0;
}

Subtask #1

100.0 / 100.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	340 KB	Output is correct
2	Correct	1 ms	336 KB	Output is correct
3	Correct	7 ms	596 KB	Output is correct
4	Correct	16 ms	932 KB	Output is correct
5	Correct	35 ms	1624 KB	Output is correct
6	Correct	59 ms	2536 KB	Output is correct
7	Correct	111 ms	4552 KB	Output is correct
8	Correct	167 ms	7048 KB	Output is correct
9	Correct	215 ms	9892 KB	Output is correct
10	Correct	342 ms	10932 KB	Output is correct