Submission #1344630

#TimeUsernameProblemLanguageResultExecution timeMemory
1344630avighnaEvent Hopping 2 (JOI21_event2)C++20
32 / 100
3093 ms15824 KiB
#include <bits/stdc++.h>

using namespace std;

struct interval {
  int l, r;
  int i;
  auto operator<=>(const interval &r) const = default;
};

class lazy_segment_tree {
  int n;
  vector<int> seg, lazy;

public:
  lazy_segment_tree(int n) : n(n), seg(4 * n), lazy(8 * n) {}

  void push(int v, int tl, int tr) {
    seg[v] += lazy[v] * (tr - tl + 1);
    lazy[2 * v] += lazy[v];
    lazy[2 * v + 1] += lazy[v];
    lazy[v] = 0;
  }

  void apply(int v, int tl, int tr, int l, int r, int x) {
    push(v, tl, tr);
    if (tr < l || r < tl) {
      return;
    }
    if (l <= tl && tr <= r) {
      lazy[v] += x;
      push(v, tl, tr);
      return;
    }
    int tm = (tl + tr) / 2;
    apply(2 * v, tl, tm, l, r, x);
    apply(2 * v + 1, tm + 1, tr, l, r, x);
    seg[v] = seg[2 * v] + seg[2 * v + 1];
  }
  void apply(int l, int r, int x) { apply(1, 0, n - 1, l, r, x); }

  int query(int v, int tl, int tr, int l, int r) {
    push(v, tl, tr);
    if (tr < l || r < tl) {
      return 0;
    }
    if (l <= tl && tr <= r) {
      return seg[v];
    }
    int tm = (tl + tr) / 2;
    return query(2 * v, tl, tm, l, r) + query(2 * v + 1, tm + 1, tr, l, r);
  }
  int query(int l, int r) { return query(1, 0, n - 1, l, r); }
};

int main() {
  cin.tie(nullptr)->sync_with_stdio(false);

  int n, k;
  cin >> n >> k;
  vector<interval> a(n);
  vector<int> buff;
  for (int i = 0, l, r; i < n; ++i) {
    cin >> l >> r;
    a[i] = {l, r, i};
    buff.push_back(l), buff.push_back(r);
  }
  auto A = a;
  sort(a.begin(), a.end(), [](interval a, interval b) { return a.r < b.r; });

  sort(buff.begin(), buff.end());
  buff.erase(unique(buff.begin(), buff.end()), buff.end());
  auto comp = [&](int i) { return lower_bound(buff.begin(), buff.end(), i) - buff.begin(); };

  auto solve = [&](int L, int R) {
    vector<interval> A;
    for (auto &[l, r, i] : a) {
      if (L <= l && r <= R) {
        A.emplace_back(l, r, i);
      }
    }
    const int n = A.size();
    if (n == 0) {
      return 0;
    }
    int cur_time = 0, ans = 0;
    for (int i = 0; i < n; ++i) {
      if (A[i].l < cur_time) {
        continue;
      }
      ans++;
      cur_time = A[i].r;
    }
    return ans;
  };

  int cur = solve(1, int(1e9));
  if (k > cur) {
    cout << "-1\n";
    return 0;
  }

  set<pair<int, int>> curs;
  lazy_segment_tree st(buff.size());
  vector<int> ans;
  for (int i = 0; i < n; ++i) {
    pair<int, int> _p = {A[i].l, int(1e9) + 1};
    auto itt = curs.upper_bound(_p);
    if (itt != curs.begin()) {
      itt--;
      // find an inteval with l <= our l
      if (A[i].r <= itt->second) { // full intersection
        continue;
      }
    }
    pair<int, int> p = {A[i].l, A[i].r};
    int q1 = st.query(comp(A[i].l), comp(A[i].r));
    int q2 = st.query(comp(A[i].l), comp(A[i].l));
    int q3 = st.query(comp(A[i].r), comp(A[i].r));
    if (q1 - q2 - q3 != 0 || q2 > 1 || q3 > 1) {
      continue;
    }
    curs.insert(p);
    auto it = curs.find(p);
    int lef = 1, rig = 1e9;
    if (it != curs.begin()) {
      lef = prev(it)->second;
    }
    if (it != --curs.end()) {
      rig = next(it)->first;
    }
    int pcur = cur;
    cur -= solve(lef, rig);
    cur += solve(lef, p.first) + 1 + solve(p.second, rig);
    if (cur < k) {
      curs.erase(it);
      cur = pcur;
    } else {
      ans.push_back(i + 1);
      st.apply(comp(p.first), comp(p.second), 1);
    }
  }

  for (int i = 0; i < k; ++i) {
    cout << ans[i] << '\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...