結果

問題 No.924 紲星
コンテスト
ユーザー lumc_
提出日時 2019-09-07 02:38:21
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 3,053 bytes
コンパイル時間 1,483 ms
コンパイル使用メモリ 117,640 KB
実行使用メモリ 39,176 KB
最終ジャッジ日時 2024-09-15 05:32:37
合計ジャッジ時間 7,469 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 5 TLE * 1 -- * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#if 0

TLEになるの？みたいな解

#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<ll> ans(q);
  for(int i = 0; i < q; i++) {
    int m = a[v[ok[i]].second];
    ll ans = 0;
    for(int j = l[i]; j <= r[i]; j++) {
      ans += abs(m - a[j]);
    }
    cout << ans << "\n";
  }
  return 0;
}
0