結果

問題 No.877 Range ReLU Query
ユーザー saxofone111saxofone111
提出日時 2021-03-18 17:31:50
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 334 ms / 2,000 ms
コード長 4,731 bytes
コンパイル時間 2,732 ms
コンパイル使用メモリ 216,228 KB
実行使用メモリ 15,696 KB
最終ジャッジ日時 2024-04-25 22:30:18
合計ジャッジ時間 7,499 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 3 ms
6,940 KB
testcase_02 AC 4 ms
6,944 KB
testcase_03 AC 4 ms
6,940 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 4 ms
6,944 KB
testcase_09 AC 2 ms
6,940 KB
testcase_10 AC 3 ms
6,940 KB
testcase_11 AC 323 ms
14,020 KB
testcase_12 AC 286 ms
15,108 KB
testcase_13 AC 231 ms
12,032 KB
testcase_14 AC 246 ms
11,312 KB
testcase_15 AC 334 ms
14,652 KB
testcase_16 AC 319 ms
14,948 KB
testcase_17 AC 319 ms
14,764 KB
testcase_18 AC 317 ms
15,696 KB
testcase_19 AC 321 ms
15,248 KB
testcase_20 AC 324 ms
14,732 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include "bits/stdc++.h"

#define MOD 1000000007
#define rep(i, n) for(ll i=0; i < (n); i++)
#define rrep(i, n) for(ll i=(n)-1; i >=0; i--)
#define ALL(v) v.begin(),v.end()
#define rALL(v) v.rbegin(),v.rend()
#define FOR(i, j, k) for(ll i=j;i<k;i++)
#define debug_print(var) cerr << #var << "=" << var <<endl;
#define DUMP(i, v)for(ll i=0;i<v.size();i++)cerr<<v[i]<<" "
#define fi first
#define se second

using namespace std;
typedef long long int ll;
typedef vector<ll> llvec;
typedef vector<double> dvec;
typedef pair<ll, ll> P;
typedef long double ld;
struct edge{ll x, c;};

/*
  struct segment_tree{
  ll N;
  llvec v;
  ll init=5e18;//initial value
  ll f(ll a, ll b){ //function
  return min(a, b);
  }
  segment_tree(ll n){
  N=1;
  while(N<n){
  N*=2;
  }
  v = llvec(2*N-1, init);
  }
  
  void set(ll i, ll val){
  i += N-1;
  v[i] = val;
  while(i>0){
  i = (i-1)/2;
  v[i] = f(v[i*2+1], v[i*2+2]);
  }
  }
  void add(ll i, ll val){
  i += N-1;
  v[i] += val;
  while(i>0){
  i = (i-1)/2;
  v[i] = f(v[i*2+1], v[i*2+2]);
  }
  }

  ll get(ll L, ll R){// L <= i < R
  L += N-1;
  R += N-1;
  ll vl = init;
  ll vr = init;
  while(L<R){
  if(L%2==0){
  vl = f(vl, v[L]);
  L++;
  }
  if(R%2==0){
  vr = f(vr, v[R-1]);
  R--;
  }
  R=(R-1)/2;
  L=(L-1)/2;
  }
  return f(vl, vr);
  }

  ll operator[](ll i){
  return v[i+N-1];
  }
  
  };*/


template <class S, S (*op)(S, S), S (*e)()> struct segtree {
public:
  int ceil_pow2(int n) {
    int x = 0;
    while ((1U << x) < (unsigned int)(n)) x++;
    return x;
  }
  segtree() : segtree(0) {}
  segtree(int n) : segtree(std::vector<S>(n, e())) {}
  segtree(const std::vector<S>& v) : _n(int(v.size())) {
    log = ceil_pow2(_n);
    size = 1 << log;
    d = std::vector<S>(2 * size, e());
    for (int i = 0; i < _n; i++) d[size + i] = v[i];
    for (int i = size - 1; i >= 1; i--) {
      update(i);
    }
  }

  void set(int p, S x) {
    assert(0 <= p && p < _n);
    p += size;
    d[p] = x;
    for (int i = 1; i <= log; i++) update(p >> i);
  }

  S get(int p) {
    assert(0 <= p && p < _n);
    return d[p + size];
  }

  S prod(int l, int r) {
    assert(0 <= l && l <= r && r <= _n);
    S sml = e(), smr = e();
    l += size;
    r += size;

    while (l < r) {
      if (l & 1) sml = op(sml, d[l++]);
      if (r & 1) smr = op(d[--r], smr);
      l >>= 1;
      r >>= 1;
    }
    return op(sml, smr);
  }

  S all_prod() { return d[1]; }

  template <bool (*f)(S)> int max_right(int l) {
    return max_right(l, [](S x) { return f(x); });
  }
  template <class F> int max_right(int l, F f) {
    assert(0 <= l && l <= _n);
    assert(f(e()));
    if (l == _n) return _n;
    l += size;
    S sm = e();
    do {
      while (l % 2 == 0) l >>= 1;
      if (!f(op(sm, d[l]))) {
        while (l < size) {
          l = (2 * l);
          if (f(op(sm, d[l]))) {
            sm = op(sm, d[l]);
            l++;
          }
        }
        return l - size;
      }
      sm = op(sm, d[l]);
      l++;
    } while ((l & -l) != l);
    return _n;
  }

  template <bool (*f)(S)> int min_left(int r) {
    return min_left(r, [](S x) { return f(x); });
  }
  template <class F> int min_left(int r, F f) {
    assert(0 <= r && r <= _n);
    assert(f(e()));
    if (r == 0) return 0;
    r += size;
    S sm = e();
    do {
      r--;
      while (r > 1 && (r % 2)) r >>= 1;
      if (!f(op(d[r], sm))) {
        while (r < size) {
          r = (2 * r + 1);
          if (f(op(d[r], sm))) {
            sm = op(d[r], sm);
            r--;
          }
        }
        return r + 1 - size;
      }
      sm = op(d[r], sm);
    } while ((r & -r) != r);
    return 0;
  }

private:
  int _n, size, log;
  std::vector<S> d;

  void update(int k) { d[k] = op(d[2 * k], d[2 * k + 1]); }
};

using S = P;

S op(S a, S b){
  return {a.fi+b.fi, a.se+b.se};
}
S e(){
  return {0, 0};
}

//segtree<S, op, e> sg(vector<S> a)

using PP = pair<P, P>;

/**************************************
** A main function starts from here  **
***************************************/
int main(){
  ll N, Q;
  cin >> N >> Q;
  vector<P> a(N);
  rep(i, N){
    ll t;
    cin >> t;
    a[i]={t, i};
  }
  sort(rALL(a));
  
  vector<S> tmp(N, {0, 0});
  segtree<S, op, e> sg(tmp);

  vector<PP> query;
  rep(i, Q){
    ll l, r, x;
    ll t;
    cin >>t>> l >> r >> x;
    l--;
    query.push_back({{x, i}, {l, r}});
  }

  llvec ans(Q, 0);


  sort(rALL(query));
  ll l = 0;
  
  for(auto iq: query){
    while(l<N and iq.fi.fi<a[l].fi){
      sg.set(a[l].se, {a[l].fi, 1});
      //cerr << l << " " << a[l].fi <<" "<< a[l].se<<endl;
      l++;
    }
    P s = sg.prod(iq.se.fi, iq.se.se);
    ans[iq.fi.se] = s.fi - s.se*iq.fi.fi;
  }

  rep(i, Q){
    cout << ans[i]<<endl;
  }
  return 0;
}
0