結果

問題 No.649 ここでちょっとQK！
ユーザー face4face4
提出日時 2020-01-29 15:34:11
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 186 ms / 3,000 ms
コード長 3,434 bytes
コンパイル時間 957 ms
コンパイル使用メモリ 85,916 KB
実行使用メモリ 12,928 KB
最終ジャッジ日時 2024-09-16 01:02:30
合計ジャッジ時間 6,152 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 38 ms
5,376 KB
testcase_04 AC 96 ms
12,928 KB
testcase_05 AC 97 ms
12,928 KB
testcase_06 AC 97 ms
12,928 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 75 ms
7,424 KB
testcase_13 AC 75 ms
7,424 KB
testcase_14 AC 74 ms
7,424 KB
testcase_15 AC 78 ms
7,424 KB
testcase_16 AC 76 ms
7,936 KB
testcase_17 AC 84 ms
8,064 KB
testcase_18 AC 101 ms
8,448 KB
testcase_19 AC 106 ms
8,832 KB
testcase_20 AC 117 ms
9,088 KB
testcase_21 AC 128 ms
9,472 KB
testcase_22 AC 142 ms
9,856 KB
testcase_23 AC 149 ms
10,368 KB
testcase_24 AC 160 ms
10,624 KB
testcase_25 AC 172 ms
11,008 KB
testcase_26 AC 186 ms
11,392 KB
testcase_27 AC 3 ms
5,376 KB
testcase_28 AC 2 ms
5,376 KB
testcase_29 AC 3 ms
5,376 KB
testcase_30 AC 67 ms
7,296 KB
testcase_31 AC 70 ms
7,680 KB
testcase_32 AC 2 ms
5,376 KB
testcase_33 AC 2 ms
5,376 KB
testcase_34 AC 2 ms
5,376 KB
testcase_35 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<random>
#include<vector>
using namespace std;
typedef long long ll;

// 0-indexed
template<class T>
struct Treap{
    struct Node{
        T val;
        int pri;
        int cnt;
        Node* ch[2];
        Node(T v, int p) : val(v), pri(p){
            cnt = 1;
            ch[0] = ch[1] = NULL;
        }
    };

    Node* root;
    vector<T> v;    // 列を返すとき用
    T def;  // !!!

    Treap(T d) : def(d){
        root = NULL;
    }

    int count(Node* t)  { return t==NULL ? 0 : t->cnt;}

    Node* update(Node* t){
        t->cnt = count(t->ch[0]) + count(t->ch[1]) + 1;
        return t;
    }

    Node* rotate(Node* t, int b){
        Node* s = t->ch[b];
        t->ch[b] = s->ch[1-b];
        s->ch[1-b] = t;
        update(t);
        update(s);
        return s;
    }

    Node* insert(Node* t, int k, T val, int pri){
        if(t == NULL)   return new Node(val, pri);
        int leftCnt = count(t->ch[0]);
        int b = k > leftCnt;
        t->ch[b] = insert(t->ch[b], k-(b ? leftCnt+1 : 0), val, pri);
        update(t);
        if(t->ch[b]->pri > t->pri)  t = rotate(t, b);
        return t;
    }

    Node* erase(Node* t, int k, bool me=false){
        if(!me && count(t) <= k)   return t;    // out-of-range
        me |= count(t->ch[0])==k;
        if(me){
            if(t->ch[0]==NULL && t->ch[1]==NULL){
                delete t;
                return NULL;
            }
            int b = (t->ch[0]==NULL ? -1 : t->ch[0]->pri) < (t->ch[1]==NULL ? -1 : t->ch[1]->pri);
            t = rotate(t, b);
            t->ch[1-b] = erase(t->ch[1-b], k, me);
        }else{
            if(count(t->ch[0]) > k){
                t->ch[0] = erase(t->ch[0], k, me);
            }else{  // definitely count(t->ch[0]) < k
                t->ch[1] = erase(t->ch[1], k-count(t->ch[0])-1, me);
            }
        }
        return update(t);
    }

    T get(Node* t, int k){
        int leftCnt = count(t->ch[0]);
        if(leftCnt > k)     return get(t->ch[0], k);
        if(leftCnt == k)    return t->val;
        return get(t->ch[1], k-1-leftCnt);
    }

    Node* insert(int k, T val){
        return root = insert(root, k, val, rand());
    }

// !!! 雑拡張
    Node* insert_value(Node* t, T val, int pri){
        if(t == NULL)   return new Node(val, pri);
        int b = t->val < val;
        t->ch[b] = insert_value(t->ch[b], val, pri);
        update(t);
        if(t->pri < t->ch[b]->pri)  t = rotate(t, b);
        return t;
    }
    Node* insert_value(T val){
        return root = insert_value(root, val, rand());
    }
// !!!

    Node* erase(int k){
        return root = erase(root, k);
    }
    T get(int k){
        return k < count(root) ? get(root, k) : def;    // !!!
    }
    
    void flatten(Node* t){
        if(t==NULL) return;
        if(t->ch[0] != NULL)    flatten(t->ch[0]);
        v.push_back(t->val);
        if(t->ch[1] != NULL)    flatten(t->ch[1]);
    }

    vector<T> flatten(){
        v.clear();
        flatten(root);
        return v;
    }
};

int main(){
    Treap<ll> t(-1);
    int q, k;
    scanf("%d %d", &q, &k);
    k--;
    while(q--){
        int op;
        ll x;
        scanf("%d", &op);
        if(op == 1){
            scanf("%lld", &x);
            t.insert_value(x);
        }else if(op == 2){
            printf("%lld\n", t.get(k));
            if(t.get(k) != -1)  t.erase(k);
        }
    }
    return 0;
}
0