Submission #1087137

#	Time	Username	Problem	Language	Result	Execution time	Memory
1087137		avighna	Lasers (NOI19_lasers)	C++17	41 / 100	108 ms	262144 KiB

lasers

#include <bits/stdc++.h>

typedef long long ll;

class FenwickTree {
public:
  std::vector<ll> f;
  ll n;

  FenwickTree(ll n) {
    this->n = n;
    f.resize(n + 1);
  }

  void add(ll idx, ll del) {
    for (ll i = idx; i <= n; i += i & (-i)) {
      f[i] += del;
    }
  }

  ll query(ll idx) {
    ll ans = 0;
    for (ll i = idx; i >= 1; i -= i & (-i)) {
      ans += f[i];
    }
    return ans;
  }
  ll query(ll l, ll r) {
    if (l > r) {
      return 0;
    }
    ll ans = query(r);
    if (l - 1 >= 1) {
      ans -= query(l - 1);
    }
    return ans;
  }
};

int main() {
  std::ios_base::sync_with_stdio(false);
  std::cin.tie(nullptr);

  ll l, r;
  std::cin >> l >> r;
  std::priority_queue<std::pair<ll, ll>, std::vector<std::pair<ll, ll>>,
                      std::greater<>>
      pq;
  std::vector<ll> left(r), sum(r);
  std::vector<std::deque<ll>> rows(r);
  FenwickTree tree(ll(1e6 + 100));
  // x blocks from right ==> l - x + 1 from left
  auto f = [&](ll cnt) { return l - (sum[cnt] - left[cnt]) + 1; };
  for (ll cnt = 0; cnt < r; ++cnt) {
    auto &v = rows[cnt];
    ll len;
    std::cin >> len;
    v.resize(len);
    for (auto &i : v) {
      std::cin >> i;
    }
    sum[cnt] = std::accumulate(v.begin(), v.end(), 0LL);
    pq.push({left[cnt] + v[0], cnt});
    v.pop_front();
    tree.add(f(cnt), 1);
  }

  ll ans = 0;
  for (ll _i = 1; _i <= l; ++_i) {
    while (!pq.empty() and pq.top().first < _i) {
      auto [l, i] = pq.top();
      pq.pop();
      tree.add(f(i), -1);
      left[i] = l;
      tree.add(f(i), 1);
      if (!rows[i].empty()) {
        pq.push({left[i] + rows[i][0], i});
        rows[i].pop_front();
      }
    }
    ll gs = tree.query(_i + 1, l + 1);
    // for (ll j = 0; j < r; ++j) {
    //   gs += _i < l - (sum[j] - left[j]) + 1;
    // }
    ans += gs != r;
  }
  std::cout << ans << '\n';
}

• Subtask 1: 0 / 10
• Subtask 2: 0 / 14
• Subtask 3: 20 / 20
• Subtask 4: 21 / 21
• Subtask 5: 0 / 22
• Subtask 6: 0 / 13