# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
857369 | resting | Overtaking (IOI23_overtaking) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#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 fix(vector<pair<int,int>> &shit){
vector<pair<int,int>> res;
sort(shit.begin(), shit.end());
for(int i = shit.size()-1; i >= 0; i--){
if(shits[i].first >= res.back().first) continue;
res.push_back(shits[i]);
}
reverse(res.begin(), res.end());
swap(shit, res);
}
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;
fix(l1);
int p2 = l2.size() - 1;
fix(l2);
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());
// cout<<"restriction"<<endl;
// for(auto &x : restriction) cout<<x.first<<" "<<x.second<<endl;
}
fix(restriction);
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;
}