結果

問題 No.649 ここでちょっとQK！
ユーザー ats5515ats5515
提出日時 2018-02-09 22:52:35
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 280 ms / 3,000 ms
コード長 2,931 bytes
コンパイル時間 753 ms
コンパイル使用メモリ 86,248 KB
実行使用メモリ 12,800 KB
最終ジャッジ日時 2024-04-17 05:36:47
合計ジャッジ時間 5,154 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
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 280 ms
5,376 KB
testcase_04 AC 96 ms
12,800 KB
testcase_05 AC 93 ms
12,672 KB
testcase_06 AC 96 ms
12,672 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 1 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 80 ms
7,424 KB
testcase_13 AC 80 ms
7,552 KB
testcase_14 AC 78 ms
7,552 KB
testcase_15 AC 78 ms
7,552 KB
testcase_16 AC 76 ms
7,552 KB
testcase_17 AC 87 ms
8,064 KB
testcase_18 AC 96 ms
8,448 KB
testcase_19 AC 106 ms
8,704 KB
testcase_20 AC 114 ms
9,216 KB
testcase_21 AC 127 ms
9,728 KB
testcase_22 AC 140 ms
10,112 KB
testcase_23 AC 151 ms
10,368 KB
testcase_24 AC 194 ms
11,008 KB
testcase_25 AC 178 ms
11,264 KB
testcase_26 AC 185 ms
11,776 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 2 ms
5,376 KB
testcase_29 AC 2 ms
5,376 KB
testcase_30 AC 69 ms
7,552 KB
testcase_31 AC 71 ms
7,552 KB
testcase_32 AC 2 ms
5,376 KB
testcase_33 AC 1 ms
5,376 KB
testcase_34 AC 2 ms
5,376 KB
testcase_35 AC 1 ms
5,376 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In member function ‘avl_tree<T>::node* avl_tree<T>::rank(avl_tree<T>::node*, long long int) const [with T = long long int]’:
main.cpp:89:9: warning: control reaches end of non-void function [-Wreturn-type]
   89 |         }
      |         ^

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string>
#include <iomanip>
#include <algorithm>
#include <cmath>
#include <stdio.h>
using namespace std;
#define int long long
int MOD = 1000000007;
template <class T>
struct avl_tree {
	struct node {
		T key;
		int size, height;
		node *child[2];
		node(const T &key) : key(key), size(1), height(1) {
			child[0] = child[1] = 0;
		}
	} *root;
	typedef node *pointer;
	avl_tree() { root = NULL; }

	pointer find(const T &key) { return find(root, key); }
	node *find(node *t, const T &key) {
		if (t == NULL) return NULL;
		if (key == t->key) return t;
		else if (key < t->key) return find(t->child[0], key);
		else                   return find(t->child[1], key);
	}
	void insert(const T &key) { root = insert(root, new node(key)); }
	node *insert(node *t, node *x) {
		if (t == NULL) return x;
		if (x->key <= t->key) t->child[0] = insert(t->child[0], x);
		else                  t->child[1] = insert(t->child[1], x);
		t->size += 1;
		return balance(t);
	}
	void erase(const T &key) { root = erase(root, key); }
	node *erase(node *t, const T &x) {
		if (t == NULL) return NULL;
		if (x == t->key) {
			return move_down(t->child[0], t->child[1]);
		}
		else {
			if (x < t->key) t->child[0] = erase(t->child[0], x);
			else            t->child[1] = erase(t->child[1], x);
			t->size -= 1;
			return balance(t);
		}
	}
	node *move_down(node *t, node *rhs) {
		if (t == NULL) return rhs;
		t->child[1] = move_down(t->child[1], rhs);
		return balance(t);
	}
#define sz(t) (t ? t->size : 0)
#define ht(t) (t ? t->height : 0)
	node *rotate(node *t, int l, int r) {
		node *s = t->child[r];
		t->child[r] = s->child[l];
		s->child[l] = balance(t);
		if (t) t->size = sz(t->child[0]) + sz(t->child[1]) + 1;
		if (s) s->size = sz(s->child[0]) + sz(s->child[1]) + 1;
		return balance(s);
	}
	node *balance(node *t) {
		for (int i = 0; i < 2; ++i) {
			if (ht(t->child[!i]) - ht(t->child[i]) < -1) {
				if (ht(t->child[i]->child[!i]) - ht(t->child[i]->child[i]) > 0)
					t->child[i] = rotate(t->child[i], i, !i);
				return rotate(t, !i, i);
			}
		}
		if (t) t->height = max(ht(t->child[0]), ht(t->child[1])) + 1;
		if (t) t->size = sz(t->child[0]) + sz(t->child[1]) + 1;
		return t;
	}
	pointer rank(int k) const { return rank(root, k); }
	pointer rank(node *t, int k) const {
		if (!t) return NULL;
		int m = sz(t->child[0]);
		if (k  < m) return rank(t->child[0], k);
		if (k == m) return t;
		if (k  > m) return rank(t->child[1], k - m - 1);
	}
};
signed main() {
	cin.tie(0);
	ios::sync_with_stdio(false);
	int Q, K;
	cin >> Q >> K;
	avl_tree<int> t;
	for (int i = 0; i < Q; i++) {
		int a;
		cin >> a;
		if (a == 1) {
			cin >> a;
			t.insert(a);
		}
		else {
			auto p = t.rank(K - 1);
			if (p == NULL) {
				cout << -1 << endl;
			}
			else {
				cout << (*p).key << endl;
				t.erase((*p).key);
			}
		}
	}
	//cout << res << endl;
}
0