Submission #858648

#TimeUsernameProblemLanguageResultExecution timeMemory
858648restingOvertaking (IOI23_overtaking)C++17
100 / 100
1434 ms161432 KiB
#include <bits/stdc++.h>

using namespace std;
#define int long long

#define ff first
#define ss second

set<pair<int, int>> restriction; // if "starts after a", has to "end after b" yk

int x, l;
void init(int32_t L, int32_t N, std::vector<long long> T, std::vector<int32_t> W, int32_t X, int32_t M, std::vector<int32_t> S)
{
    restriction = {};
    x = X; l = L;
    vector<vector<pair<int, int>>> restrictions(M);// yes
    vector<pair<int, int>> cur;
    for (int i = 0; i < N; i++) if (W[i] > X) cur.push_back({ T[i], W[i] - X });
    for (int i = 0; i < M-1; i++) {
        sort(cur.begin(), cur.end());
        vector<pair<int, int>> res;
        const int T = cur.size();
        for (int j = 0; j < T; j++) 
            res.push_back({ cur[j].ff + cur[j].ss * (S[i + 1] - S[i]), cur[j].ss });
        for (int j = 0; j < T - 1; j++) 
            if (res[j].ff > res[j + 1].ff) 
                res[j + 1].ff = res[j].ff;
        int p  = -1;
        for (int j = 0; j < T; j++) {
            if(res[j].ff != p){
                if(j + 1 == T || cur[j].ff != cur[j+1].ff){
                    p = res[j].ff;
                    restrictions[i].push_back({ cur[j].ff+1, res[j].ff }); //start, end
                }
            }
        }

        swap(res, cur);
    }
    for (int i = 0; i < M; i++) {
        reverse(restrictions[i].begin(), restrictions[i].end());
        for(auto x : restrictions[i]){
            auto y = restriction.upper_bound({x.second, numeric_limits<long long>::max()});
            if(y == restriction.begin()) {
                if(restriction.size() && y->second <= x.first){
                    continue;
                }
                restriction.insert({x.second, x.first});
                continue;
            }
            else y = prev(y);
            int tt = x.first;
            while(y != restriction.end() && y->first >= x.first){
                tt = min(tt, y->second);
                auto t = y;
                if(t == restriction.begin()) t= restriction.end();
                else t = prev(t);
                restriction.erase(y);
                y = t;
            }
            x.first = tt;
            auto z = restriction.upper_bound({x.second, numeric_limits<long long>::max()});
            if(z != restriction.end() && z->second <= x.first) continue;
            restriction.insert({x.second, x.first});
        }
    }
    set<pair<int,int>> bruh;
    for(auto &x : restriction) bruh.insert({x.second, x.first});
    restriction=bruh;
    return;
}

long long arrival_time(long long Y)
{
    auto yes = restriction.upper_bound(make_pair( Y, numeric_limits<long long>::max() ));
    if(yes == restriction.begin()) yes = restriction.end();
    else yes = prev(yes);
    if (yes != restriction.end()) Y = max(Y, yes->second);
    return Y + x * l;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...