結果

問題 No.789 範囲の合計
ユーザー face4face4
提出日時 2019-02-08 22:44:56
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 171 ms / 1,000 ms
コード長 1,483 bytes
コンパイル時間 996 ms
コンパイル使用メモリ 89,432 KB
実行使用メモリ 11,136 KB
最終ジャッジ日時 2024-07-01 11:40:16
合計ジャッジ時間 3,364 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 167 ms
9,728 KB
testcase_03 AC 70 ms
6,940 KB
testcase_04 AC 171 ms
10,496 KB
testcase_05 AC 146 ms
9,388 KB
testcase_06 AC 151 ms
9,856 KB
testcase_07 AC 63 ms
6,944 KB
testcase_08 AC 158 ms
11,136 KB
testcase_09 AC 148 ms
10,624 KB
testcase_10 AC 148 ms
7,832 KB
testcase_11 AC 137 ms
10,368 KB
testcase_12 AC 132 ms
10,368 KB
testcase_13 AC 2 ms
6,944 KB
testcase_14 AC 2 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<map>
#include<vector>
#include<algorithm>
#include<set>
using namespace std;
typedef long long ll;

struct BIT{
    // 1-origin
    vector<ll> v;
    int n;

    BIT(int _n):
        v(vector<ll>(_n+1, 0)), n(_n) {}

    ll sum(int i){
        ll s = 0;
        while(i > 0){
            s += v[i];
            i -= lsb(i);
        }
        return s;
    }

    ll sum(int l, int r){
        return sum(r) - sum(l-1);
    }

    void add(int i, int x){
        while(i <= n){
            v[i] += x;
            i += lsb(i);
        }
    }

private:
    // least significant bit
    int lsb(int i){
        return i & -i;
    }
};

int main(){
    int n;
    cin >> n;

    set<int> s;
    int a[n], b[n], c[n];
    for(int i = 0; i < n; i++){
        cin >> a[i] >> b[i] >> c[i];
        if(a[i] == 0)   s.insert(b[i]);
    }

    s.insert(0);
    s.insert(1000000000);
    map<int,int> zip;
    int cur = 1;
    for(auto it = s.begin(); it != s.end(); it++){
        zip[*it] = cur++;
    }

    BIT bit(zip.size());

    ll ans = 0;
    for(int i = 0; i < n; i++){
        if(a[i] == 0){
            bit.add(zip[b[i]], c[i]);
        }else if(a[i] == 1){
            auto l = zip.lower_bound(b[i]);
            auto r = zip.lower_bound(c[i]);
            int le = (*l).second, ri = (*r).second;
            if((*r).first != c[i])  ri--;
            if(le <= ri)    ans += bit.sum(le, ri);
        }
    }

    cout << ans << endl;
    
    return 0;
}
0