結果

問題 No.877 Range ReLU Query
ユーザー MayimgMayimg
提出日時 2019-09-07 09:54:59
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 200 ms / 2,000 ms
コード長 4,754 bytes
コンパイル時間 2,755 ms
コンパイル使用メモリ 223,808 KB
実行使用メモリ 19,048 KB
最終ジャッジ日時 2024-11-08 10:14:09
合計ジャッジ時間 6,032 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 3 ms
5,248 KB
testcase_02 AC 3 ms
5,248 KB
testcase_03 AC 3 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 3 ms
5,248 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 3 ms
5,248 KB
testcase_11 AC 186 ms
17,772 KB
testcase_12 AC 167 ms
16,740 KB
testcase_13 AC 124 ms
13,160 KB
testcase_14 AC 131 ms
13,540 KB
testcase_15 AC 175 ms
18,664 KB
testcase_16 AC 196 ms
18,260 KB
testcase_17 AC 196 ms
18,532 KB
testcase_18 AC 200 ms
18,404 KB
testcase_19 AC 168 ms
19,044 KB
testcase_20 AC 192 ms
19,048 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define _USE_MATH_DEFINES
#include <bits/stdc++.h>
using namespace std;
template<typename Monoid> class SegmentTree {
private:
  using Func = function<Monoid(Monoid, Monoid)>;
  Func func;
  Monoid unity;
  int siz;
  vector<Monoid> data;
  int n;
  int height;
  inline void build() {
    siz = 1;
    while (siz < n) siz <<= 1, ++height;
    data.assign(siz << 1, unity);
    for (int i = 0; i < n; i++) set(i, unity);
    for (int idx = siz - 1; idx > 0; idx--) data[idx] = func(data[(idx << 1) | 0], data[(idx << 1) | 1]);
  }
  template<typename T> inline void build(const T& arr) {
    if (!~n) n = (int) arr.size();
    siz = 1;
    while (siz < n) siz <<= 1, ++height;
    data.assign(siz << 1, unity);
    for (int i = 0; i < n; i++) set(i, arr[i]);
    for (int idx = siz - 1; idx > 0; idx--) data[idx] = func(data[(idx << 1) | 0], data[(idx << 1) | 1]);
  }
  inline void set(int idx, const Monoid& val) { assert(idx + siz < (int) data.size()); data[idx + siz] = val;}

public:
  SegmentTree() {}
  SegmentTree (const Func f, const Monoid& u, const int& n_) : func(f), unity(u), n(n_) { build(); }
  template<typename T> SegmentTree (const T& arr, const Func f, const Monoid& u, int n_ = -1) : func(f), unity(u), n(n_) { build(arr); }
  

  void initialize (const Func f, const Monoid &u, const int& n_) { func = f; unity = u; n = n_; build(); }
  template<typename T> void initialize (const T& arr, const Func f, const Monoid& u, int n_ = -1) { func = f; unity = u; n = n_; buld(arr); }

  void modify (int idx, const Monoid& val) {
    idx += siz;
    data[idx] = val;
    while (idx >>= 1) data[idx] = func(data[(idx << 1) | 0], data[(idx << 1) | 1]);
  }
  //[left, right) 
  Monoid get (int left, int right) {
    if (left > right) swap(left, right);
    Monoid val_left = unity, val_right = unity;
    for (int l = left + siz, r = right + siz; l < r; l >>= 1, r >>= 1) {
      if (l & 1) val_left = func(val_left, data[l++]);
      if (r & 1) val_right = func(data[--r], val_right);
    }
    return func(val_left, val_right);
  }

  template<typename U>
  int find (int start, int left, int right, int cur, Monoid& val, const U& f) {
    if (left + 1 == right) {
      val = func(val, data[cur]);
      return f(val) ? cur - siz : -1;
    }
    int mid = (left + right) >> 1;
    if (mid <= start) return find(start, mid, right, (cur << 1) | 1, val, f);
    if (start <= left && !f(func(val, data[cur]))) {
      val = func(val, data[cur]);
      return -1;
    }
    int val_left = find(start, left, mid, (cur << 1) | 0, val, f);
    if (~val_left) return val_left;
    return find(start, mid, right, (cur << 1) | 1, val, f);
  }
  template<typename U>
  int find (int start, const U& f) {
    Monoid val = unity;
    return find(start, 0, siz, 1, val, f);
  }

  inline Monoid operator[] (int idx) { return data[idx + siz]; }
  void print() {
    for (int i = 0; i < siz; i++) {
      cout << (*this)[i];
      if (i != siz - 1) cout << ", ";
    }
    cout << '\n';
  }
};

// long long max
function<long long(long long, long long)> LLMAX = [] (long long a, long long b) -> long long { return max(a, b); };
const long long LLMAX_UNITY = numeric_limits<long long>::min();
// int max
function<int(int, int)> INTMAX = [] (int a, int b) -> int { return max(a, b); };
const int INTMAX_UNITY = numeric_limits<int>::min();
// long long min
function<long long(long long, long long)> LLMIN = [] (long long a, long long b) -> long long { return min(a, b); };
const long long LLMIN_UNITY = numeric_limits<long long>::max();
// int min
function<int(int, int)> INTMIN = [] (int a, int b) -> int { return min(a, b); };
const int INTMIN_UNITY = numeric_limits<int>::max();
// long long sum
function<long long(long long, long long)> LLSUM = [] (long long a, long long b) -> long long { return a + b; };
const long long LLSUM_UNITY = 0LL;
// int sum
function<int(int, int)> INTSUM = [] (int a, int b) -> int { return a + b; };
const int INTSUM_UNITY = 0;

signed main() { 
  ios::sync_with_stdio(false); cin.tie(0);
  int n, q;
  cin >> n >> q;
  vector<vector<int>> qry;
  for (int i = 0; i < n; i++) {
    int a;
    cin >> a;
    qry.push_back({a, 0, i});
  }
  for (int i = 0; i < q; i++) {
    int c, l, r, x;
    cin >> c >> l >> r >> x;
    l--;
    qry.push_back({x, c, l, r, i});
  }
  sort(qry.rbegin(), qry.rend());
  SegmentTree<long long> active(LLSUM, LLSUM_UNITY, n);
  SegmentTree<long long> sgt(LLSUM, LLSUM_UNITY, n);
  vector<long long> ans(q);
  for (auto  v : qry) {
    if (v[1] == 0) {
      active.modify(v[2], 1);
      sgt.modify(v[2], v[0]);
    } else {
      ans[v.back()] = sgt.get(v[2], v[3]) - 1LL * v[0] * active.get(v[2], v[3]);
    }
  }
  for (int i = 0; i < q; i++) {
    cout << ans[i] << '\n';
  }
  return 0;
}
0