結果

問題 No.924 紲星
ユーザー lumc_lumc_
提出日時 2019-09-06 00:25:00
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 483 ms / 4,000 ms
コード長 3,396 bytes
コンパイル時間 1,386 ms
コンパイル使用メモリ 118,412 KB
実行使用メモリ 35,212 KB
最終ジャッジ日時 2023-10-13 08:19:49
合計ジャッジ時間 6,925 ms
ジャッジサーバーID
(参考情報)
judge14 / judge12
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,352 KB
testcase_01 AC 1 ms
4,348 KB
testcase_02 AC 1 ms
4,348 KB
testcase_03 AC 2 ms
4,352 KB
testcase_04 AC 1 ms
4,352 KB
testcase_05 AC 2 ms
4,352 KB
testcase_06 AC 2 ms
4,352 KB
testcase_07 AC 2 ms
4,356 KB
testcase_08 AC 483 ms
34,032 KB
testcase_09 AC 464 ms
33,724 KB
testcase_10 AC 473 ms
33,840 KB
testcase_11 AC 474 ms
35,212 KB
testcase_12 AC 472 ms
33,544 KB
testcase_13 AC 200 ms
18,136 KB
testcase_14 AC 191 ms
17,592 KB
testcase_15 AC 178 ms
16,720 KB
testcase_16 AC 214 ms
19,272 KB
testcase_17 AC 291 ms
25,032 KB
testcase_18 AC 1 ms
4,348 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#if 0

おそらく AC 解
assert 除去

#endif
// includes {{{
#include<iostream>
#include<iomanip>
#include<algorithm>
#include<vector>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<tuple>
#include<cmath>
#include<random>
#include<cassert>
#include<bitset>
#include<cstdlib>
// #include<deque>
// #include<multiset>
// #include<cstring>
// #include<bits/stdc++.h>
// }}}
using namespace std;
using ll = long long;

// .add(i, v)   : bit[i] += v
// .get(i)      : bit[i]
// .sum(i)      : bit[0] + ... + bit[i]
// .range(l, r) : bit[l] + ... + bit[r]
// .lower_bound(T v) : min i that satisfies .sum(i) >= v
//    - use only when bit[i] >= 0 for all i > 0
/// --- Binary Indexed Tree {{{ ///
#include <cassert>
#include <vector>
template < class T = long long >
struct BinaryIndexedTree {
  using size_type = std::size_t;
  size_type n, m;
  T identity;
  std::vector< T > data;
  BinaryIndexedTree() : n(0) {}
  BinaryIndexedTree(int n, T identity = T())
    : n(n), identity(identity), data(n, identity) {
      m = 1;
      while(m < n) m <<= 1;
    }
  void add(size_type i, T x) {
    assert(i < n);
    i++;
    while(i <= n) {
      data[i - 1] = data[i - 1] + x;
      i += i & -i;
    }
  }
  T sum(int i) {
    if(i < 0) return identity;
    if(i >= n) i = n - 1;
    i++;
    T s = identity;
    while(i > 0) {
      s = s + data[i - 1];
      i -= i & -i;
    }
    return s;
  }
  T get(int i) { return sum(i) - sum(i - 1); }
  T range(int a, int b) { return sum(b) - sum(a - 1); }
  size_type lower_bound(T w) {
    size_type i = 0;
    for(size_type k = m; k > 0; k >>= 1) {
      if(i + k <= n && data[i + k - 1] < w) w -= data[(i += k) - 1];
    }
    return i;
  }
};
/// }}}--- ///

template < class T = long long >
using BIT = BinaryIndexedTree< T >;

int main() {
  std::ios::sync_with_stdio(false), std::cin.tie(0);
  int n, q;
  cin >> n >> q;

  vector<ll> a(n);
  vector<pair<ll, int>> v;
  for(int i = 0; i < n; i++) {
    cin >> a[i], v.emplace_back(a[i], i);
  }
  sort(begin(v), end(v));
  vector<int> l(q), r(q);
  for(int i = 0; i < q; i++) {
    cin >> l[i] >> r[i];
    l[i]--, r[i]--;
  }

  vector<int> ok(q, n-1);
  vector<int> ng(q, -1);
  if(n > 1) {
    vector<vector<int>> mid(n);
    mid[n/2].reserve(n);
    for(int i = 0; i < q; i++) mid[n/2].push_back(i);
    // パラサーチ O(Q log N) * O(log N)
    int rest = q;
    while(rest) {
      BIT<> bit(n);
      for(int i = 0; i < n; i++) {
        int id = v[i].second;
        bit.add(id, 1);
        for(int j : mid[i]) {
          if(bit.range(l[j], r[j]) >= (r[j] - l[j] + 2) / 2) ok[j] = i;
          else ng[j] = i;
          if(abs(ok[j] - ng[j]) > 1) {
            mid[(ok[j] + ng[j])/2].push_back(j);
          } else {
            rest--;
          }
        }
        mid[i].clear();
      }
    }
  }

  vector<vector<int>> qs(n);
  for(int i = 0; i < q; i++) qs[ok[i]].push_back(i);
  // dump(ok);

  BIT<> d1(n), d2(n);
  vector<ll> ans(q);
  for(int i = 0; i < n; i++) d2.add(i, a[i]);
  for(int i = 0; i < n; i++) {
    int id = v[i].second;
    d2.add(id, -a[id]);
    d1.add(id, a[id]);

    for(auto j : qs[i]) {
      int len = r[j] - l[j] + 1;
      ans[j] = - d1.range(l[j], r[j]) + d2.range(l[j], r[j]);
      if(len % 2 == 1) ans[j] += a[id];
      // dump(j, len, a[id]);
    }

  }
  for(int i = 0; i < q; i++) cout << ans[i] << "\n";
  return 0;
}
0