Submission #1294217

#	Time	Username	Problem	Language	Result	Execution time	Memory
1294217		avighna	Pairs (IOI07_pairs)	C++20	100 / 100	2410 ms	193472 KiB

pairs

#pragma GCC optimize("Ofast", "unroll-all-loops")

#include <algorithm>
#include <iostream>
#include <unordered_map>
#include <vector>

template <typename T, int d, bool sparse>
class fenwick_tree;
template <typename T, bool sparse>
class fenwick_tree<T, 1, sparse> {
private:
  using storage_t = std::conditional_t<sparse, std::unordered_map<std::size_t, T>, std::vector<T>>;
  std::size_t n;
  storage_t f;

public:
  fenwick_tree(std::size_t n) : n(n) {
    if constexpr (!sparse) {
      f.assign(n + 1, T{});
    }
  }
  void apply(std::size_t i, T x) {
    for (++i; i <= n; i += i & -i) {
      f[i] = f[i] + x;
    }
  }
  T query(std::size_t i) {
    T ans = T{};
    for (++i; i > 0; i -= i & -i) {
      if constexpr (sparse) {
        if (f.contains(i)) {
          ans = ans + f.at(i);
        }
      } else {
        ans = ans + f[i];
      }
    }
    return ans;
  }
  T query(std::size_t l, std::size_t r)
    requires requires(T a, T b) { a - b; }
  {
    return l == 0 ? query(r) : query(r) - query(l - 1);
  }
};
template <typename T, int d, bool sparse>
class fenwick_tree {
private:
  std::size_t n;
  std::vector<fenwick_tree<T, d - 1, sparse>> f;

public:
  template <typename... ints>
  fenwick_tree(std::size_t n, ints... dims) : n(n), f(n + 1, fenwick_tree<T, d - 1, sparse>{dims...}) {}
  template <typename... ints>
  void apply(std::size_t i, ints... next) {
    for (++i; i <= n; i += i & -i) {
      f[i].apply(next...);
    }
  }
  template <typename... ints>
    requires(sizeof...(ints) == d - 1)
  T query(std::size_t i, ints... dims) {
    T ans = T{};
    for (++i; i > 0; i -= i & -i) {
      ans = ans + f[i].query(dims...);
    }
    return ans;
  }
  template <typename... ints>
  T query(std::size_t l, std::size_t r, ints... dims)
    requires requires(T a, T b) { a - b; }
  {
    T ans = T{};
    for (++r; r > 0; r -= r & -r) {
      ans = ans + f[r].query(dims...);
    }
    for (; l > 0; l -= l & -l) {
      ans = ans - f[l].query(dims...);
    }
    return ans;
  }
};

using namespace std;

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

  int b, n, d, m;
  cin >> b >> n >> d >> m;
  m++;
  int64_t ans = 0;
  if (b == 1) {
    vector<int> a(n);
    for (int &i : a) {
      cin >> i;
    }
    sort(a.begin(), a.end());
    for (int i = 0, j = 0; i < n; ++i) {
      while (j < n && a[j] - a[i] <= d) {
        j++;
      }
      ans += j - i - 1;
    }
  } else if (b == 2) {
    fenwick_tree<int, 2, true> fenwick(2 * m, 3 * m);
    for (int i = 0, u, v; i < n; ++i) {
      cin >> u >> v;
      int x = v - u + m, y = v + u + m;
      ans += fenwick.query(max(0, x - d), min(x + d, 2 * m - 1), max(0, y - d), min(y + d, 3 * m - 1));
      fenwick.apply(x, y, 1);
    }
  } else {
    fenwick_tree<int, 3, false> fenwick(m, 2 * m, 3 * m);
    for (int i = 0, u, v, w; i < n; ++i) {
      cin >> u >> v >> w;
      int x = w, y = v - u + m, z = v + u + m;
      for (int dx = -d; dx <= d; ++dx) {
        int dd = d - abs(dx);
        if (dd >= 0 && x + dx >= 0 && x + dx < m) {
          ans += fenwick.query(x + dx, x + dx, max(0, y - dd), min(y + dd, 2 * m - 1), max(0, z - dd), min(z + dd, 3 * m - 1));
        }
      }
      fenwick.apply(x, y, z, 1);
    }
  }

  cout << ans << '\n';
}

Compilation message (stderr)

pairs.cpp: In instantiation of 'fenwick_tree<T, d, sparse>::fenwick_tree(std::size_t, ints ...) [with ints = {int}; T = int; int d = 2; bool sparse = true; std::size_t = long unsigned int]':
pairs.cpp:109:52:   required from here
pairs.cpp:55:93: warning: narrowing conversion of 'dims#0' from 'int' to 'std::size_t' {aka 'long unsigned int'} [-Wnarrowing]
   55 |   fenwick_tree(std::size_t n, ints... dims) : n(n), f(n + 1, fenwick_tree<T, d - 1, sparse>{dims...}) {}
      |                                                                                             ^~~~
pairs.cpp: In instantiation of 'fenwick_tree<T, d, sparse>::fenwick_tree(std::size_t, ints ...) [with ints = {int, int}; T = int; int d = 3; bool sparse = false; std::size_t = long unsigned int]':
pairs.cpp:117:56:   required from here
pairs.cpp:55:93: warning: narrowing conversion of 'dims#0' from 'int' to 'std::size_t' {aka 'long unsigned int'} [-Wnarrowing]
pairs.cpp: In instantiation of 'fenwick_tree<T, d, sparse>::fenwick_tree(std::size_t, ints ...) [with ints = {int}; T = int; int d = 2; bool sparse = false; std::size_t = long unsigned int]':
pairs.cpp:55:101:   required from 'fenwick_tree<T, d, sparse>::fenwick_tree(std::size_t, ints ...) [with ints = {int, int}; T = int; int d = 3; bool sparse = false; std::size_t = long unsigned int]'
pairs.cpp:117:56:   required from here
pairs.cpp:55:93: warning: narrowing conversion of 'dims#0' from 'int' to 'std::size_t' {aka 'long unsigned int'} [-Wnarrowing]

• Subtask 1: 6 / 6
• Subtask 2: 6 / 6
• Subtask 3: 6 / 6
• Subtask 4: 6 / 6
• Subtask 5: 6 / 6
• Subtask 6: 7 / 7
• Subtask 7: 7 / 7
• Subtask 8: 8 / 8
• Subtask 9: 8 / 8
• Subtask 10: 10 / 10
• Subtask 11: 10 / 10
• Subtask 12: 10 / 10
• Subtask 13: 10 / 10