#115628 (C++14) No.151 セグメントフィッシング

• 問題: No.151 セグメントフィッシング
• ユーザー: izuru_matsuura
• 提出日時: 2016-09-09 12:05:18
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 140 ms / 5,000 ms
• コード長: 4,966 bytes
• コンパイル時間: 1,802 ms
• コンパイル使用メモリ: 176,632 KB
• 実行使用メモリ: 6,820 KB
• 最終ジャッジ日時: 2024-11-16 05:18:14
• 合計ジャッジ時間: 3,988 ms
• ジャッジサーバーID （参考情報）: judge2 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,820 KB
testcase_01	AC	2 ms; 6,816 KB
testcase_02	AC	2 ms; 6,816 KB
testcase_03	AC	2 ms; 6,820 KB
testcase_04	AC	2 ms; 6,816 KB
testcase_05	AC	2 ms; 6,816 KB
testcase_06	AC	2 ms; 6,816 KB
testcase_07	AC	2 ms; 6,820 KB
testcase_08	AC	23 ms; 6,816 KB
testcase_09	AC	23 ms; 6,816 KB
testcase_10	AC	23 ms; 6,820 KB
testcase_11	AC	23 ms; 6,820 KB
testcase_12	AC	106 ms; 6,820 KB
testcase_13	AC	106 ms; 6,816 KB
testcase_14	AC	105 ms; 6,816 KB
testcase_15	AC	106 ms; 6,816 KB
testcase_16	AC	105 ms; 6,816 KB
testcase_17	AC	52 ms; 6,820 KB
testcase_18	AC	60 ms; 6,820 KB
testcase_19	AC	140 ms; 6,820 KB
testcase_20	AC	140 ms; 6,816 KB
testcase_21	AC	85 ms; 6,820 KB
testcase_22	AC	84 ms; 6,820 KB
testcase_23	AC	55 ms; 6,816 KB

ソースコード

#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;
            V l = (t->left == NULL ? 0 : t->left->sum);
            V r = (t->right == NULL ? 0 : t->right->sum);
            return l + r + 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)の分
                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));
            }
        }
    }
}

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