結果

問題 No.1170 Never Want to Walk
ユーザー zeronosu77108
提出日時 2020-08-14 23:03:03
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,411 bytes
コンパイル時間 1,151 ms
コンパイル使用メモリ 132,060 KB
実行使用メモリ 22,872 KB
最終ジャッジ日時 2024-10-10 16:26:51
合計ジャッジ時間 10,877 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 4 WA * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;
using int64 = long long;

struct LazySegmentTree {
private:
    int64 n;
    vector<int64> node, lazy;

public:
    LazySegmentTree(vector<int64> v) {
        int64 sz = (int64)v.size();
        n = 1; while(n < sz) n *= 2;
        node.resize(2*n-1);
        lazy.resize(2*n-1, 0);

        for(int64 i=0; i<sz; i++) node[i+n-1] = v[i];
        for(int64 i=n-2; i>=0; i--) node[i] = node[i*2+1] + node[i*2+2];
    }

    // k 番目のノードについて遅延評価を行う
    void eval(int64 k, int64 l, int64 r) {

        // 遅延配列が空でない場合、自ノード及び子ノードへの
        // 値の伝播が起こる
        if(lazy[k] != 0) {
            node[k] += lazy[k];

            // 最下段かどうかのチェックをしよう
            // 子ノードは親ノードの 1/2 の範囲であるため、
            // 伝播させるときは半分にする
            if(r - l > 1) {
                lazy[2*k+1] += lazy[k] / 2;
                lazy[2*k+2] += lazy[k] / 2;
            }

            // 伝播が終わったので、自ノードの遅延配列を空にする
            lazy[k] = 0;
        }
    }

    void add(int64 a, int64 b, int64 x, int64 k=0, int64 l=0, int64 r=-1) {
        if(r < 0) r = n;

        // k 番目のノードに対して遅延評価を行う
        eval(k, l, r);

        // 範囲外なら何もしない
        if(b <= l || r <= a) return;

        // 完全に被覆しているならば、遅延配列に値を入れた後に評価
        if(a <= l && r <= b) {
            lazy[k] += (r - l) * x;
            eval(k, l, r);
        }

            // そうでないならば、子ノードの値を再帰的に計算して、
            // 計算済みの値をもらってくる
        else {
            add(a, b, x, 2*k+1, l, (l+r)/2);
            add(a, b, x, 2*k+2, (l+r)/2, r);
            node[k] = node[2*k+1] + node[2*k+2];
        }
    }

    int64 getsum(int64 a, int64 b, int64 k=0, int64 l=0, int64 r=-1) {
        if(r < 0) r = n;
        if(b <= l || r <= a) return 0;

        // 関数が呼び出されたら評価！
        eval(k, l, r);
        if(a <= l && r <= b) return node[k];
        int64 vl = getsum(a, b, 2*k+1, l, (l+r)/2);
        int64 vr = getsum(a, b, 2*k+2, (l+r)/2, r);
        return vl + vr;
    }
};


int main() {
    int64 n,a,b;
    cin >> n >> a >> b;
    vector x(n,0LL);
    for (int64 i=0; i<n; i++) cin >> x[i];
    LazySegmentTree rseg(vector(n,0LL));
    LazySegmentTree lseg(vector(n,0LL));

    cerr<<"calc1"<<endl;
    for (int64 i=0; i<n; i++) {
        auto l = lower_bound(x.begin(), x.end(), x[i]+a);
        auto r = upper_bound(x.begin(), x.end(), x[i]+b);
        int64 lindex = l - x.begin();
        int64 rindex = r - x.begin();
        rseg.add(lindex, rindex, rindex-lindex + rseg.getsum(i,i+1));
    }
    cerr<<"calc2"<<endl;
    for (int64 i=n-1; i>=0; i--) {
        auto l = lower_bound(x.begin(), x.end(), x[i]-b);
        auto r = upper_bound(x.begin(), x.end(), x[i]-a);
        int64 lindex = l - x.begin();
        int64 rindex = r - x.begin();
        lseg.add(lindex, rindex, rindex-lindex + lseg.getsum(i,i+1));
    }


    cerr<<"ans"<<endl;
    for (int64 i=0; i<n; i++) {
        cout <<  lseg.getsum(i,i+1) + rseg.getsum(i, i+1) + 1 << endl;
    }
}
0