Submission #155538

#	Time	Username	Problem	Language	Result	Execution time	Memory
155538		Learner99	Segway (COI19_segway)	C++14	40 / 100	1056 ms	84932 KiB

segway

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp> 
#include <ext/pb_ds/tree_policy.hpp> 

using namespace std;
using namespace __gnu_pbds;  

#define int long long
#define N 100006

int a[N][3];


  
#define ordered_set tree< pair < int , int > , null_type,less< pair < int , int > >, rb_tree_tag,tree_order_statistics_node_update> 
 
ordered_set s[405];
int acc[405] = {0};
int cur[N];
int ans[N];

int32_t main()
{
    int n;
    cin>>n;

    for(int i=1;i<=n;i++)
    {
        cin>>a[i][0]>>a[i][1]>>a[i][2];
    }

    int m;
    cin>>m;
    
    for(int i=0;i<m;i++)
    {
        int temp;
        cin>>temp;
        acc[temp] = 1;
    }

    set < pair < int , int > > rid; 
    
    for(int i=1;i<=n;i++)
    {
        rid.insert({0, i});
    }
    while(rid.size()!=0)
    {
        pair < int , int > t = *rid.begin();
        rid.erase(t);
        int tme = t.first;
        int rider = t.second;

        if( acc[cur[rider]] == 0)
        {
            cur[rider]++;
            int x = 0;
            if(cur[rider]<= 100)
            {
                ans[rider]+=a[rider][0];
                x = a[rider][0];
            }
            else if(cur[rider]<=200)
            {
                ans[rider]+=a[rider][1];
                x = a[rider][1];
            }
            else
            {
                ans[rider]+=a[rider][2];
                x = a[rider][2];
            }

            if(cur[rider]<300)
            {
                rid.insert({tme+x, rider});
            }
            s[cur[rider]].insert({tme+x, rider});
        }
        else
        {
            int aage = s[cur[rider]].order_of_key({tme, 0});
            aage%=20;

            if(aage==0)
            {
                cur[rider]++;
                int x = 0;
                if(cur[rider]<= 100)
                {
                    ans[rider]+=a[rider][0];
                    x = a[rider][0];
                }
                else if(cur[rider]<=200)
                {
                    ans[rider]+=a[rider][1];
                    x = a[rider][1];
                }
                else
                {
                    ans[rider]+=a[rider][2];
                    x = a[rider][2];
                }

                if(cur[rider]<300)
                {
                    rid.insert({tme+x, rider});
                }
                s[cur[rider]].insert({tme+x, rider});
            }
            else
            {
                int y = 1;
                int p = cur[rider];
                for(int i=p+1;i<=min(300ll, p+aage);i++)
                {
                    ans[rider]++;
                    cur[rider]++;
                    s[i].insert({tme+y, rider});
                    y++;
                }

                if(cur[rider] < 300)
                    rid.insert({tme+aage, rider});
            }
        }
    }

    for(int i=1;i<=n;i++)
    {
        cout<<ans[i]<<"\n";
    }
    cout<<endl;

    return 0;
}

• Subtask 1: 0 / 15
• Subtask 2: 40 / 40
• Subtask 3: 0 / 45