Submission #1061838

#TimeUsernameProblemLanguageResultExecution timeMemory
1061838RecursiveCoSky Walking (IOI19_walk)C++17
43 / 100
237 ms22736 KiB
#include <bits/stdc++.h> #include "walk.h" using namespace std; long long min_distance(vector<int> x, vector<int> h, vector<int> l, vector<int> r, vector<int> y, int s, int g) { int n = x.size(); // = h.size() int m = l.size(); // = r.size() = y.size() if (max(n, m) <= 50) { // s and g can be anything, but n and m are small anyway so it doesn't matter. map<pair<int, int>, vector<array<int, 3>>> adjList; vector<vector<int>> pts(n); for (int i = 0; i < m; i++) { vector<int> enough; for (int j = l[i]; j <= r[i]; j++) { if (h[j] >= y[i]) { pts[j].push_back(y[i]); enough.push_back(j); } } int t = enough.size(); for (int j = 0; j < t - 1; j++) { int x1 = enough[j]; int x2 = enough[j + 1]; adjList[{x1, y[i]}].push_back({x2, y[i], x[x2] - x[x1]}); adjList[{x2, y[i]}].push_back({x1, y[i], x[x2] - x[x1]}); } } for (int i = 0; i < n; i++) { pts[i].push_back(0); sort(pts[i].begin(), pts[i].end()); int t = pts[i].size(); for (int j = 0; j < t - 1; j++) { int y1 = pts[i][j]; int y2 = pts[i][j + 1]; adjList[{i, y1}].push_back({i, y2, y2 - y1}); adjList[{i, y2}].push_back({i, y1, y2 - y1}); } } priority_queue<pair<long long, pair<int, int>>, vector<pair<long long, pair<int, int>>>, greater<pair<long long, pair<int, int>>>> pq; map<pair<int, int>, long long> dist; set<pair<int, int>> visited; pq.push({0ll, {s, 0}}); while (!pq.empty()) { auto top = pq.top(); pq.pop(); if (visited.find(top.second) != visited.end()) continue; visited.insert(top.second); dist[top.second] = top.first; for (auto con: adjList[top.second]) { if (visited.find({con[0], con[1]}) == visited.end()) { pq.push({top.first + con[2], {con[0], con[1]}}); } } } return dist.find({g, 0}) == dist.end()? -1: dist[{g, 0}]; } // Assume s = 0, g = n - 1 and h consists of equal elements. long long ans = x[n - 1] - x[0]; map<pair<int, int>, long long> dp; vector<array<int, 4>> events; for (int i = 0; i < m; i++) { events.push_back({r[i] - 1, y[i], i, 1}); events.push_back({l[i] - 1, y[i], i, 0}); } sort(events.rbegin(), events.rend()); vector<vector<int>> right(n); for (int i = 0; i < m; i++) right[r[i]].push_back(i); set<pair<int, int>> open; int ptr = 0; for (int i = n - 1; i >= 0; i--) { while (ptr < 2 * m && events[ptr][0] >= i) { int yc = events[ptr][1]; int ind = events[ptr][2]; int typ = events[ptr][3]; if (typ) { open.insert({yc, ind}); } else { open.erase({yc, ind}); } ptr++; } for (int ind: right[i]) { long long val = 1e18; auto it = open.lower_bound({y[ind], -1}); if (it != open.end()) { val = min(val, (long long) ((*it).first - y[ind]) + dp[{r[(*it).second], (*it).first}]); } if (it != open.begin()) { it--; val = min(val, (long long) (y[ind] - (*it).first) + dp[{r[(*it).second], (*it).first}]); } dp[{i, y[ind]}] = i == n - 1? y[ind]: val; } } long long extra = 1e18; for (int i = 0; i < m; i++) { if (l[i] == 0) { extra = min(extra, y[i] + dp[{r[i], y[i]}]); } } ans += extra; if (ans >= 1e18) return -1; return ans; }
#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...