結果

問題 No.877 Range ReLU Query
ユーザー gasingasin
提出日時 2019-09-07 22:24:41
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 663 ms / 2,000 ms
コード長 2,950 bytes
コンパイル時間 916 ms
コンパイル使用メモリ 79,368 KB
実行使用メモリ 102,732 KB
最終ジャッジ日時 2024-11-08 10:17:29
合計ジャッジ時間 7,872 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
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 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 3 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 4 ms
5,248 KB
testcase_09 AC 3 ms
5,248 KB
testcase_10 AC 3 ms
5,248 KB
testcase_11 AC 616 ms
82,988 KB
testcase_12 AC 531 ms
80,972 KB
testcase_13 AC 403 ms
61,264 KB
testcase_14 AC 427 ms
56,012 KB
testcase_15 AC 657 ms
100,548 KB
testcase_16 AC 646 ms
97,872 KB
testcase_17 AC 663 ms
100,048 KB
testcase_18 AC 657 ms
100,940 KB
testcase_19 AC 322 ms
102,608 KB
testcase_20 AC 451 ms
102,732 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:102:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  102 |     scanf("%d%d", &n, &q);
      |     ~~~~~^~~~~~~~~~~~~~~~
main.cpp:105:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  105 |         scanf("%lld", &a[i].first);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~
main.cpp:114:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  114 |         scanf("%lld%lld%lld%lld", &f, &l, &r, &x);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff #

#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <map>
#include <queue>
#include <cstdio>
#include <string.h>
#define rep(i,n) for (int i = 0; i < (int)n; i++)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pi;
typedef pair<pi, pi> pp;
typedef pair<ll, ll> pl;
double PI = 3.1415926535897932;
const double EPS = 1e-9;
const ll MOD = 1000000007;
const int inf = 1 << 30;
const ll linf = 1LL << 60;

pl operator+(const pl &l, const pl &r) {
    return pl(l.first+r.first, l.second+r.second);
}

// 区間和、要素更新のsegtree

template<typename T>
struct permanent_segtree {
    struct NODE {
        T val;
        struct NODE *l, *r;
    };

    int n;
    T unit;
    vector<NODE*> roots;

    void init(int n_, T unit_){
        n = 1;
        unit = unit_;
        while(n < n_) n *= 2;
        vector<NODE*> nodes;
        rep(i,2*n-1) {
            NODE* nn = new NODE();
            nn->val = unit;
            nodes.push_back(nn);
        }
        rep(i,2*n-1) {
            if (i < n-1) {
                nodes[i]->l = nodes[2*i+1];
                nodes[i]->r = nodes[2*i+2];
            }
        }
        roots.push_back(nodes[0]);
    }

    void app(int a, T x, NODE* root, int l, int r) {
        int b = a+1;
        if (a <= l && r <= b) {
            root->val = x;
            return;
        }
        if ((l+r)/2 <= a || b <= l) {
            NODE* nr = new NODE(*(root->r));
            root->r = nr;
            app(a, x, nr, (l+r)/2, r);
        } else {
            NODE* nl = new NODE(*(root->l));
            root->l = nl;
            app(a, x, nl, l, (l+r)/2);
        }
        root->val = root->r->val + root->l->val;
    }

    void update(int k, T a){
        NODE* nr = new NODE(*(roots.back()));
        roots.push_back(nr);

        app(k, a, nr, 0, n);
    }

    T sub_query(int a, int b, NODE* root, int l, int r){
        if(r <= a || b <= l) return unit;
        if(a <= l && r <= b) return root->val;

        T vl = sub_query(a,b,root->l,l,(l+r)/2);
        T vr = sub_query(a,b,root->r,(l+r)/2,r);
        return vl + vr;
    }

    T query (int a, int b, int t) {
        return sub_query(a, b, roots[t], 0, n);
    }
};

permanent_segtree<pl> seg;
int n, q;
pl a[100000];

int main() {
    scanf("%d%d", &n, &q);
    seg.init(n, pl(0,0));
    rep(i,n) {
        scanf("%lld", &a[i].first);
        a[i].second = i;
    }
    sort(a, a+n, greater<pl>());
    rep(i,n) {
        seg.update(a[i].second, pl(a[i].first, 1));
    }
    rep(i,q) {
        ll f, l, r, x;
        scanf("%lld%lld%lld%lld", &f, &l, &r, &x);
        l--; r--;
        int st = -1, en = n; 
        while (en-st > 1) {
            int mid = (st + en) / 2;
            if (a[mid].first < x) en = mid;
            else st = mid;
        }
        pl val = seg.query(l, r+1, st+1);
        printf("%lld\n", val.first-val.second*x);
    }
}
0