結果

問題 No.151 セグメントフィッシング
コンテスト
ユーザー izuru_matsuura
提出日時 2016-09-09 00:14:02
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 4,969 bytes
コンパイル時間 4,185 ms
コンパイル使用メモリ 184,240 KB
実行使用メモリ 821,668 KB
最終ジャッジ日時 2024-11-15 22:36:45
合計ジャッジ時間 94,742 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample MLE * 5
other TLE * 11 MLE * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

namespace {

    typedef double real;
    typedef long long ll;

    template<class T> ostream& operator<<(ostream& os, const vector<T>& vs) {
        if (vs.empty()) return os << "[]";
        os << "[" << vs[0];
        for (int i = 1; i < vs.size(); i++) os << " " << vs[i];
        return os << "]";
    }
    template<class T> istream& operator>>(istream& is, vector<T>& vs) {
        for (auto it = vs.begin(); it != vs.end(); it++) is >> *it;
        return is;
    }

    template<class V> struct Treap {
        mt19937 mt;
        Treap() : root(NULL) {
            random_device rnd;
            mt.seed(rnd());
        }

        struct Node {
            V value;
            Node *left, *right;
            int priority;
            int size;
            V sum; /**/
            Node(V value, int priority) : value(value), priority(priority), size(1), sum(value)/**/ {
                left = NULL;
                right = NULL;
            }
        };
        
        Node *root;
        int size(Node *t) {
            if (t == NULL) return 0;
            return t->size;
        }

        V sum(Node *t) { /**/
            if (t == NULL) return 0;
            return sum(t->left) + sum(t->right) + t->value; 
        }

        Node* update(Node *v) {
            v->size = size(v->left) + size(v->right) + 1;
            v->sum = sum(v->left) + sum(v->right) + v->value; /**/
            return v;
        }

        Node* merge(Node *l, Node *r) {
            if (l == NULL) return r;
            if (r == NULL) return l;
            assert(l != NULL && r != NULL);

            if (l->priority > r->priority) {
                l->right = merge(l->right, r);
                return update(l);
            } else {
                r->left = merge(l, r->left);
                return update(r);
            }
        }

        pair<Node*, Node*> split(Node *t, int k) { // [0, k) 縺ｨ [k, n)縺ｫ蛻・牡
            if (t == NULL) return make_pair((Node*)NULL, (Node*)NULL);
            if (k <= size(t->left)) {
                auto s = split(t->left, k);
                t->left = s.second;
                return make_pair(s.first, update(t));
            } else {
                auto s = split(t->right, k - size(t->left) - 1); // 縺薙・-1縺ｯ莉願ｦ九※繧矩Κ蛻・惠縺ｮroot(縺､縺ｾ繧閣)縺ｮ蛻・                t->right = s.first;
                return make_pair(update(t), s.second);
            }
        }
        pair<Node*, Node*> split(int k) { return split(root, k); }

        Node* insert(Node *t, int k, V value) {
            auto x = split(t, k);
            auto y = split(x.second, 1);
            return merge(x.first, merge(new Node(value, mt()), y.second));
        }
        Node* insert(int k, V value) { return root = insert(root, k, value); }

        Node* erase(Node *t, int k) {
            auto x = split(t, k);
            auto y = split(x.second, 1);
            return merge(x.first, y.second);
        }
        Node* erase(int k) { return root = erase(root, k); }

        Node* find(Node *t, int k) {
            auto x = split(t, k);
            auto y = split(x.second, 1);
            Node *ret = y.first;
            merge(x.first, merge(y.first, y.second));
            return ret;
        }
        Node* find(int k) { return find(root, k); }

    };

    void solve() {
        int N, Q; cin >> N >> Q;
        Treap<ll> l, r;
        for (int i = 0; i < N; i++) {
            l.insert(i, 0);
            r.insert(i, 0);
        }

        for (int q = 0; q < Q; q++) {
            string x; int y, z; 
            cin >> x >> y >> z;
            //cout << x << " " << y << " " << z << endl;

            auto ls = l.split(1);
            auto rs = r.split(N - 1);
            l.root = l.merge(ls.second, rs.second);
            r.root = r.merge(ls.first, rs.first);
            //cout << "L:"; for (int i = 0; i < N; i++) cout << " " << l.find(i)->value; cout << endl;
            //cout << "R:"; for (int i = 0; i < N; i++) cout << " " << r.find(i)->value; cout << endl;

            if (x == "L") {
                auto v = l.find(y);
                if (v == NULL) l.insert(y, z);
                else l.insert(y, v->value + z);
            } else if (x == "R") {
                auto v = r.find(y);
                if (v == NULL) r.insert(y, z);
                else r.insert(y, v->value + z);
            } else {
                auto ls = l.split(y);
                auto lt = l.split(ls.second, z - y);
                auto rs = r.split(y);
                auto rt = r.split(rs.second, z - y);
                cout << l.sum(lt.first) + r.sum(rt.first) << endl;
                l.merge(ls.first, l.merge(lt.first, lt.second));
                r.merge(rs.first, l.merge(rt.first, rt.second));
                //cout << "nanimosina~i" << endl;
            }
        }
    }
}

int main() {
    solve();
    return 0;
}
0