Submission #1132516

#TimeUsernameProblemLanguageResultExecution timeMemory
1132516SpyrosAlivClosing Time (IOI23_closing)C++20
29 / 100
1091 ms31968 KiB
// linear network #include <bits/stdc++.h> using namespace std; #define ll long long int n, a, b; ll k; vector<int> w; vector<ll> minX, minY; vector<vector<pair<int, int>>> tree; int solve_line() { return -1; } void dfs(int node, bool a, int par = -1, ll dep = 0LL) { if (!a) minX[node] = dep; else minY[node] = dep; for (auto [next, c]: tree[node]) { if (next == par) continue; dfs(next, a, node, dep + c); } } int solve_line(int x, int y) { int currS = 0; for (int i = x+1; i < n; i++) { minX[i] = minX[i-1] + w[i - 1]; } for (int i = x-1; i >= 0; i--) { minX[i] = minX[i+1] + w[i]; } for (int i = y+1; i < n; i++) { minY[i] = minY[i-1] + w[i - 1]; } for (int i = y-1; i >= 0; i--) { minY[i] = minY[i+1] + w[i]; } vector<ll> minBoth(n, 0); vector<ll> f, s; for (int i = 0; i < n; i++) { minBoth[i] = max(minX[i], minY[i]); } f = minX; s = minY; for (int i = 1; i < n; i++) { minX[i] = minX[i-1] + minX[i]; minY[i] = minY[i-1] + minY[i]; minBoth[i] = minBoth[i-1] + minBoth[i]; } int maxAns = 0; for (int l = 0; l <= y; l++) { for (int r = max(x, l); r < n; r++) { // (l, r) included in both int ans = (r - l + 1) * 2; ll cost = minBoth[r]; if (l > 0) cost -= minBoth[l-1]; if (y > r) { cost += minY[y] - minY[r]; ans += (y - r); } if (l > x) { cost += minX[l-1]; if (x > 0) cost -= minX[x-1]; ans += (l - x); } if (cost > k) continue; int a = min(l-1, x-1), b = max(y+1, r+1); while (cost <= k && (a >= 0 || b < n)) { int nxt = 0; if (a < 0) { nxt = minY[b]; if (b > 0) nxt -= minY[b-1]; b++; } else if (b >= n) { nxt = minX[a]; if (a > 0) nxt -= minX[a-1]; a--; } else { int opt1 = minY[b], opt2 = minX[a]; if (a > 0) opt2 -= minX[a-1]; if (b > 0) opt1 -= minY[b-1]; if (opt2 <= opt1) { a--; nxt = opt2; } else { b++; nxt = opt1; } } if (nxt + cost > k) break; cost += nxt; ans++; } maxAns = max(maxAns, ans); } } int ans = 0; vector<ll> sec; for (int i = 0; i < n; i++) { sec.push_back(min(f[i], s[i])); } sort(sec.begin(), sec.end()); for (auto cr: sec) { if (cr <= k) { k -= cr; ans++; } } maxAns = max(maxAns, ans); return maxAns; } int max_score(int sz, int x, int y, ll l, vector<int> u, vector<int> v, vector<int> weights) { n = sz; tree.clear(); tree.resize(n); k = l; w = weights; for (int i = 0; i < n-1; i++) { //int xx = u[i]; tree[u[i]].push_back({v[i], w[i]}); tree[v[i]].push_back({u[i], w[i]}); } minX.clear(); minX.assign(n, 0LL); minY.clear(); minY.assign(n, 0LL); bool line = true; for (int i = 0; i < n-1; i++) { if (v[i] != u[i] + 1) { line = false; break; } } if (line) return solve_line(x, y); else { dfs(x, false); dfs(y, true); vector<ll> costs; for (int i = 0; i < n; i++) { costs.push_back(min(minX[i], minY[i])); //costs.push_back(minY[i]); } sort(costs.begin(), costs.end()); int ans = 0; int p = (int)costs.size(); for (int i = 0; i < p; i++) { ll x = costs[i]; if (k >= x) { ans++; k -= x; } else break; } return ans; } } /* int main() { cout << max_score(6, 1, 3, 5, {0, 1, 2, 3, 4}, {1, 2, 3, 4, 5}, {10, 1, 1, 1, 1}) << "\n"; }*/
#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...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...