結果

問題 No.151 セグメントフィッシング
ユーザー izuru_matsuuraizuru_matsuura
提出日時 2016-09-09 00:30:23
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
MLE  
実行時間 -
コード長 4,919 bytes
コンパイル時間 2,105 ms
コンパイル使用メモリ 185,448 KB
実行使用メモリ 819,200 KB
最終ジャッジ日時 2024-11-15 22:52:27
合計ジャッジ時間 91,350 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 MLE -
testcase_01 MLE -
testcase_02 MLE -
testcase_03 MLE -
testcase_04 MLE -
testcase_05 MLE -
testcase_06 MLE -
testcase_07 MLE -
testcase_08 MLE -
testcase_09 MLE -
testcase_10 MLE -
testcase_11 MLE -
testcase_12 TLE -
testcase_13 TLE -
testcase_14 TLE -
testcase_15 TLE -
testcase_16 TLE -
testcase_17 TLE -
testcase_18 TLE -
testcase_19 TLE -
testcase_20 TLE -
testcase_21 TLE -
testcase_22 TLE -
testcase_23 MLE -
権限があれば一括ダウンロードができます

ソースコード

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));
            }
        }
    }
}

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

0