Submission #857365

#TimeUsernameProblemLanguageResultExecution timeMemory
857365restingOvertaking (IOI23_overtaking)C++17
19 / 100
24 ms1124 KiB
#include <bits/stdc++.h>

using namespace std;
#define int long long

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

int x, l;
bool comp(const pair<int,int>&a, const pair<int,int>&b){
    if(a.second == b.second) return a.first < b.first;
    return a.second < b.second;
}
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)
{
    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].first + cur[j].second * (S[i + 1] - S[i]), cur[j].second });
        for (int j = 0; j < T - 1; j++) if (res[j].first > res[j + 1].first) res[j + 1].first = res[j].first;
        for (int j = 0; j < T; j++) {
        //assert(res[j].first >= cur[j].first+1);
        if(!j || res[j].first != res[j-1].first);
        restrictions[i].push_back({ cur[j].first+1, res[j].first }); //start, end
        }

        //for(int j = 0; j < restrictions[i].size(); j++) cout<<restrictions[i][j].first<<","<<restrictions[i][j].second<<endl;
        //cout<<endl;
        swap(res, cur);
    }
    //restrictions i guess ["start after i has to end after j"]
    restriction = restrictions[0];
    //    cout<<"restriction"<<endl;
   // for(auto &x : restriction) cout<<x.first<<" "<<x.second<<endl;
    for (int i = 1; i < M; i++) {
        //transition from prev ???
        //two pointers
        vector<pair<int, int>> nice;
        auto l1 = restriction, l2 = restrictions[i];
        int p1 = l1.size() - 1;
        sort(l1.begin(), l1.end(), comp);
        int p2 = l2.size() - 1;
        sort(l2.begin(), l2.end(), comp);
        int cur = numeric_limits<long long>::max();
        while (p1 >= 0 || p2 >= 0) {
          //  cout<<"A: "<<p1<<","<<p2<<endl;
            if (p1 < 0) {
                nice.push_back(l2[p2]);
                cur = l2[p2].first;
                p2--;
            } else if (p2 < 0) {
                nice.push_back(l1[p1]);
                cur = l1[p1].first;
                p1--;
            } else if (l1[p1].second < l2[p2].second) {
                int a = l2[p2].first;
                while (l1[p1].second >= l2[p2].first) {
                    a = min(a, l1[p1].first);
                    p1--;
                } 
                l2[p2].first = a;
                    nice.push_back(l2[p2]);
                    cur = l2[p2].first;
                    p2--;
            } else {
                assert(l1[p1].second >= l2[p2].second);
                nice.push_back(l1[p1]);
                cur = l1[p1].first;
                p1--;
            }
            //while (p1 >= 0 && l1[p1].first >= cur) p1--;
            //while (p2 >= 0 && l2[p2].first >= cur) p2--;
        }
        swap(restriction, nice);
        //reverse(restriction.begin(), restriction.end());
        sort(restriction.begin(), restriction.end(), comp);
       // cout<<"restriction"<<endl;
  //  for(auto &x : restriction) cout<<x.first<<" "<<x.second<<endl;
    }


    return;
}

long long arrival_time(long long Y)
{
    auto yes = prev(upper_bound(restriction.begin(), restriction.end(), make_pair( Y, numeric_limits<long long>::max() )))- restriction.begin();
    if (yes >= 0 && yes <= restriction.size()) Y = max(Y, restriction[yes].second);
    return Y + x * l;
}

Compilation message (stderr)

overtaking.cpp: In function 'void init(int32_t, int32_t, std::vector<long long int>, std::vector<int>, int32_t, int32_t, std::vector<int>)':
overtaking.cpp:48:13: warning: variable 'cur' set but not used [-Wunused-but-set-variable]
   48 |         int cur = numeric_limits<long long>::max();
      |             ^~~
overtaking.cpp: In function 'long long int arrival_time(long long int)':
overtaking.cpp:92:25: warning: comparison of integer expressions of different signedness: 'long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   92 |     if (yes >= 0 && yes <= restriction.size()) Y = max(Y, restriction[yes].second);
      |                     ~~~~^~~~~~~~~~~~~~~~~~~~~
#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...