Submission #1302143

#TimeUsernameProblemLanguageResultExecution timeMemory
1302143daotuankhoiEvent Hopping 2 (JOI21_event2)C++20
100 / 100
124 ms24136 KiB
#include <bits/stdc++.h> using namespace std; #define fi first #define se second #define ll long long #ifdef LOCAL #include "algo/debug.h" #else #define debug(...) 42 #endif template <class T> bool ckmax(T &a, T b) { return a < b ? (a = b, true) : false; } template <class T> bool ckmin(T &a, T b) { return a > b ? (a = b, true) : false; } const int MAXN = 2e5 + 5; int n, k; pair<int, int> e[MAXN]; namespace sub1 { int bit[MAXN], dp[MAXN]; vector<int> comp; void add(int i, int v) { for (; i > 0; i -= i & -i) ckmax(bit[i], v); } int get(int i) { int ans = 0; for (; i <= (int)comp.size(); i += i & -i) ckmax(ans, bit[i]); return ans; } void solve() { for (int i = 1; i <= n; i++) { comp.emplace_back(e[i].fi); comp.emplace_back(e[i].se); } sort(comp.begin(), comp.end()); comp.resize(unique(comp.begin(), comp.end()) - comp.begin()); for (int i = 1; i <= n; i++) { e[i].fi = lower_bound(comp.begin(), comp.end(), e[i].fi) - comp.begin() + 1; e[i].se = lower_bound(comp.begin(), comp.end(), e[i].se) - comp.begin() + 1; } for (int i = n; i >= 1; i--) { dp[i] = get(e[i].se) + 1; add(e[i].fi, dp[i]); } vector<int> ans; for (int i = 1; i <= n; i++) { if (get(e[i].se) >= k - (int)ans.size() - 1 && (ans.empty() || e[ans.back()].se <= e[i].fi)) ans.emplace_back(i); if ((int)ans.size() == k) break; } if (ans.size() < k) cout << -1 << '\n'; else for (int x : ans) cout << x << '\n'; } } namespace sub4 { int m; int bit[MAXN]; pair<int, int> b[MAXN]; vector<int> comp; int minPos[MAXN]; const int LG = 18; int up[MAXN][LG + 1]; void add(int i, int v) { for (; i > 0; i -= i & -i) ckmin(bit[i], v); } int get(int i) { int ans = m + 1; for (; i <= m; i += i & -i) ckmin(ans, bit[i]); return ans; } int countSegment(int l, int r) { int ans = 0; for (int i = LG; i >= 0; i--) { if (up[l][i] <= r) { ans |= (1 << i); l = up[l][i]; } } return ans; } void solve() { for (int i = 1; i <= n; i++) { comp.emplace_back(e[i].fi); comp.emplace_back(e[i].se); } sort(comp.begin(), comp.end()); comp.resize(unique(comp.begin(), comp.end()) - comp.begin()); m = comp.size(); for (int i = 1; i <= n; i++) { e[i].fi = lower_bound(comp.begin(), comp.end(), e[i].fi) - comp.begin() + 1; e[i].se = lower_bound(comp.begin(), comp.end(), e[i].se) - comp.begin() + 1; b[i] = e[i]; debug(b[i]); } sort(b + 1, b + n + 1); for (int i = 1; i <= m; i++) bit[i] = minPos[i] = m + 1; for (int i = 1; i <= n; i++) { ckmin(minPos[b[i].fi], b[i].se); } up[m + 1][0] = m + 1; for (int i = m; i >= 1; i--) { add(i, minPos[i]); up[i][0] = get(i); } for (int j = 1; j <= LG; j++) { for (int i = 1; i <= m + 1; i++) { up[i][j] = up[up[i][j - 1]][j - 1]; } } set<pair<int, int>> seg; if (countSegment(1, m) < k) { cout << -1 << '\n'; return; } seg.emplace(0, 1); seg.emplace(m, m); int cntChose = 0, curValid = countSegment(1, m); for (int i = 1; i <= n; i++) { if (cntChose >= k) break; auto it = seg.lower_bound({e[i].fi, 0}); if (it->fi < e[i].se) continue; if (prev(it)->se > e[i].fi) continue; int tmp = curValid; tmp -= countSegment(prev(it)->se, it->fi); tmp += countSegment(prev(it)->se, e[i].fi) + countSegment(e[i].se, it->fi); if (tmp >= k - cntChose - 1) { cntChose++; curValid = tmp; cout << i << '\n'; seg.insert(e[i]); } } } } int main() { #define NAME "test" if (fopen(NAME".inp", "r")) { freopen(NAME".inp", "r", stdin); freopen(NAME".out", "w", stdout); } ios_base::sync_with_stdio(false); cin.tie(nullptr); cin >> n >> k; bool checkSub1 = 1; for (int i = 1; i <= n; i++) { cin >> e[i].fi >> e[i].se; // debug(e[i].fi, e[i - 1].fi, e[i].fi < e[i - 1].fi); if (i > 1 && e[i].fi < e[i - 1].fi) checkSub1 = 0; } if (checkSub1) sub1::solve(); else sub4::solve(); return 0; }

Compilation message (stderr)

event2.cpp: In function 'int main()':
event2.cpp:152:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  152 |         freopen(NAME".inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
event2.cpp:153:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  153 |         freopen(NAME".out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...