結果
問題 | No.649 ここでちょっとQK! |
ユーザー | ikd |
提出日時 | 2019-04-18 15:08:08 |
言語 | C (gcc 12.3.0) |
結果 |
AC
|
実行時間 | 328 ms / 3,000 ms |
コード長 | 2,931 bytes |
コンパイル時間 | 598 ms |
コンパイル使用メモリ | 31,764 KB |
実行使用メモリ | 11,296 KB |
最終ジャッジ日時 | 2023-10-22 08:14:52 |
合計ジャッジ時間 | 6,376 ms |
ジャッジサーバーID (参考情報) |
judge10 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,348 KB |
testcase_01 | AC | 1 ms
4,348 KB |
testcase_02 | AC | 1 ms
4,348 KB |
testcase_03 | AC | 23 ms
4,348 KB |
testcase_04 | AC | 101 ms
11,296 KB |
testcase_05 | AC | 105 ms
11,296 KB |
testcase_06 | AC | 102 ms
11,296 KB |
testcase_07 | AC | 1 ms
4,348 KB |
testcase_08 | AC | 1 ms
4,348 KB |
testcase_09 | AC | 1 ms
4,348 KB |
testcase_10 | AC | 1 ms
4,348 KB |
testcase_11 | AC | 1 ms
4,348 KB |
testcase_12 | AC | 133 ms
6,144 KB |
testcase_13 | AC | 130 ms
6,144 KB |
testcase_14 | AC | 127 ms
6,132 KB |
testcase_15 | AC | 141 ms
6,148 KB |
testcase_16 | AC | 129 ms
6,148 KB |
testcase_17 | AC | 150 ms
6,568 KB |
testcase_18 | AC | 169 ms
6,988 KB |
testcase_19 | AC | 188 ms
7,400 KB |
testcase_20 | AC | 204 ms
7,828 KB |
testcase_21 | AC | 225 ms
8,248 KB |
testcase_22 | AC | 251 ms
8,672 KB |
testcase_23 | AC | 261 ms
9,096 KB |
testcase_24 | AC | 316 ms
9,520 KB |
testcase_25 | AC | 312 ms
9,940 KB |
testcase_26 | AC | 328 ms
10,360 KB |
testcase_27 | AC | 1 ms
4,348 KB |
testcase_28 | AC | 2 ms
4,348 KB |
testcase_29 | AC | 2 ms
4,348 KB |
testcase_30 | AC | 127 ms
6,140 KB |
testcase_31 | AC | 129 ms
6,136 KB |
testcase_32 | AC | 1 ms
4,348 KB |
testcase_33 | AC | 1 ms
4,348 KB |
testcase_34 | AC | 1 ms
4,348 KB |
testcase_35 | AC | 1 ms
4,348 KB |
ソースコード
#include <stdio.h> #include <stdlib.h> unsigned int rand_int(void) { static unsigned int x = 123456789, y = 362436069, z = 521288629, w = 88675123; unsigned int t; t = (x ^ (x << 11)); x = y; y = z; z = w; return (w = (w ^ (w >> 19)) ^ (t ^ (t >> 8))); } typedef long long i64; typedef struct Node { i64 val; int size; struct Node *left; struct Node *right; } Node; Node *create_node(i64 val) { Node *ret = (Node *)malloc(sizeof(Node)); ret->val = val; ret->size = 1; ret->left = ret->right = NULL; return ret; } typedef struct Pair { Node *left, *right; } Pair; Pair make_pair(Node *left, Node *right) { Pair pair = {left, right}; return pair; } int size(Node *node) { return node == NULL ? 0 : node->size; } Node *update(Node *node) { node->size = 1 + size(node->left) + size(node->right); return node; } Node *merge(Node *left, Node *right) { if (left == NULL) { return right; } else if (right == NULL) { return left; } if (rand_int() % (size(left) + size(right)) < (unsigned int)size(left)) { left->right = merge(left->right, right); return update(left); } else { right->left = merge(left, right->left); return update(right); } } Pair split(Node *root, int k) { if (root == NULL) return make_pair(root, root); if (k <= size(root->left)) { Pair nodes = split(root->left, k); root->left = nodes.right; return make_pair(nodes.left, update(root)); } else { Pair nodes = split(root->right, k - size(root->left) - 1); root->right = nodes.left; return make_pair(update(root), nodes.right); } } int count_less(Node *root, i64 x) { if (root == NULL) return 0; if (root->val < x) { return count_less(root->right, x) + size(root->left) + 1; } else { return count_less(root->left, x); } } Node *insert(Node *root, i64 x) { Pair nodes = split(root, count_less(root, x)); return merge(merge(nodes.left, create_node(x)), nodes.right); } Node *erase(Node *root, i64 x) { Pair nodes = split(root, count_less(root, x)); if (nodes.right != NULL) { return merge(nodes.left, split(nodes.right, 1).right); } else { return merge(nodes.left, nodes.right); } } Node *get_kth(Node *root, int k) { if (root == NULL || size(root) < k) return NULL; int left_size = size(root->left); if (left_size + 1 == k) { return root; } else if (left_size >= k) { return get_kth(root->left, k); } else { return get_kth(root->right, k - left_size - 1); } } int main(void) { Node *root = NULL; int q, k; scanf("%d %d", &q, &k); while (q--) { int t; scanf("%d", &t); if (t == 1) { i64 v; scanf("%lld", &v); root = insert(root, v); } else { Node *kth = get_kth(root, k); if (kth == NULL) { printf("-1\n"); } else { printf("%lld\n", kth->val); root = erase(root, kth->val); } } } return 0; }