#525468 (Java) No.649 ここでちょっとQK！

提出ソース
結果

問題	No.649 ここでちょっとQK！
ユーザー	k_6101
提出日時	2020-08-07 01:01:33
言語	Java (openjdk 23)
結果	AC
実行時間	586 ms / 3,000 ms
コード長	7,179 bytes
コンパイル時間	2,428 ms
コンパイル使用メモリ	88,996 KB
実行使用メモリ	70,556 KB
最終ジャッジ日時	2024-09-22 05:14:54
合計ジャッジ時間	14,136 ms
ジャッジサーバーID （参考情報）	judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン	結果
sample	AC * 4
other	AC * 32
権限があれば一括ダウンロードができます
ソースコード

raw source code
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.net.ConnectException;
import java.rmi.dgc.Lease;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Deque;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.NoSuchElementException;
import java.util.Objects;
import java.util.Set;
import java.util.Stack;
import java.util.TreeMap;
import java.util.TreeSet;

import static java.util.Comparator.*;

public class Main {
	public static void main(String[] args) {
		Main main = new Main();
		main.solve();
		main.out.close();
	}
	// ======================================================================
	BIT bit = new BIT(200010);
	Compress comp = new Compress();
	public void solve() {
		int Q = ni();
		int K = ni();
		int[] t = new int[Q];
		long[] v = new long[Q];
		for (int q = 0; q < Q; q++) {
			t[q] = ni();
			if(t[q] == 1) {
				v[q] = nl();
			}
		}
		// 座標圧縮
		comp.compress(v);
		int size, cnt;
		for (int i = 0; i < Q; i++) {
			if(t[i] == 1)	bit.add(comp.getNo(i)+1, 1);
			else {
				size = bit.sum(200000);
				if(size < K)	out.println(-1);
				else {
					cnt = bit.lower_bound(K);
					out.println(comp.getValue(cnt-1));
					bit.add(cnt, -1);
				}
			}
		}
	}
	// 座標圧縮
	class Compress {
		Set<Long> set = new TreeSet();
		long[] sortedValue_;	// 元の値（インデックスは圧縮後のソートされた位置）
		int[] originalNo_;		// 圧縮後のソートされた位置（インデックスは元の位置）

		// 圧縮する
		public void compress(long[] arg) {
			// 並べ替えて、重複削除
			for (int i = 0; i < arg.length; i++) {
				set.add(arg[i]);
			}
			sortedValue_ = new long[set.size()];
			int cnt = 0;
			for(long val : set)	{
				sortedValue_[cnt++] = val;
			}
			originalNo_ = new int[arg.length];
			for (int i = 0; i < arg.length; i++) {
				originalNo_[i] = Arrays.binarySearch(sortedValue_, arg[i]);
			}
		}
		// 圧縮後の位置を返す
		public int getNo(int index) {
			return originalNo_[index];
		}
		// 元の値を返す
		public long getValue(int index) {
			return sortedValue_[index];
		}
	}

	// Binary Indexed Tree (Fenwick Tree)
	// 数列の初めの要素からi番目までの区間和を求めるために使います
	//
	// 存在する数の位置に　１　を足していけば、X が何番目に小さいかは先頭から X までの合計となる
	// これを２分探索すると Y 番目に小さい数を取得できる
	class BIT {
		int size_;
		int[] data_;
		
		// 初期化でサイズを指定する
		public BIT(int n) {
			size_ = n;						// データサイズ
			data_ = new int[size_+1];		// １～なので　＋１する
		}
		// 指定したインデックス（１～）に指定した数を足す  O(logN)
		public void add(int index, int x) {
			if(index == 0) {
				out.println("インデックス　０　は範囲外です");
				System.exit(-1);
			}
			// LSB(Least Significant Bit) = (i & -i)
			// LSBとは二進数で表現した時に初めて1が出てくるのは右から数えて何番目かというもの
			// LSBを足すことで加算する先を決める事ができる
			for (int i = index; i <= size_; i += (i & -i)) {
				data_[i] += x;
			}
		}
		// １から指定したインデックス（１～）までの合計を計算する  O(logN)
		public int sum(int r) {			// data_[1] +   + data_[r]
			int ret = 0;		
			// LSBを引くことで加算するデータを決める事ができる
			for (int i = r; i > 0; i -= (i & -i)) {
				ret += data_[i];
			}
			return ret;
		}
		// インデックス　ｌ　から　ｒ　までの合計を計算する
		public int sum(int l, int r) {	// v[l] +   + v[r]
			return sum(r-1) - sum(l-1);
		}
		// 二分探索
		// １から指定したインデックス（１～）までの合計が
		// 引数で渡される値以上となる最小インデックスを求める　→　なければ　０
		// ☆　前提条件　data_[*] >= 0
		public int lower_bound(long val) {
			if(val <= 0)	return 0;
			int ans = 0, r = 1;
			// r = size_ 以下の最大の２のべき乗
			while(r < size_)	r = r << 1;
			for (int i = r; i > 0; i /= 2) {
				if(ans + i <= size_ && data_[ans + i] < val) {
					val -= data_[ans + i];
					ans += i;
				}
			}
			return ans + 1;
		}
	}

	// ------------------------------------------
	// ライブラリ
	// ------------------------------------------
	// Print
	private PrintWriter out = new PrintWriter(System.out);
	
	// Scanner
	private FastScanner scan = new FastScanner();
	int ni(){ return scan.nextInt();}
	int[] ni(int n){int[] a = new int[n]; for(int i = 0; i < n; i++){a[i] = ni();} return a;}
	int[][] ni(int y, int x){int[][] a = new int[y][x];
	for(int i = 0; i < y; i++){for(int j = 0; j < x; j++){a[i][j] = ni();}} return a;}
	long nl(){return scan.nextLong();}
	long[] nl(int n){long[] a = new long[n]; for(int i = 0; i < n; i++){a[i] = nl();} return a;}
	long[][] nl(int y, int x){long[][] a = new long[y][x];
	for(int i = 0; i < y; i++){for(int j = 0; j < x; j++){a[i][j] = nl();}} return a;}
	String ns(){return scan.next();}
	String[] ns(int n){String[] a = new String[n]; for(int i = 0; i < n; i++){a[i] = ns();} return a;}
	String[][] ns(int y, int x){String[][] a = new String[y][x];
	for(int i = 0; i < y; i++){for(int j = 0; j < x; j++){a[i][j] = ns();}} return a;}
}
class FastScanner {
	private final InputStream in = System.in;
	private final byte[] buffer = new byte[1024];
	private int ptr = 0;
	private int buflen = 0;
	private boolean hasNextByte() {
		if (ptr < buflen) {
			return true;
		}else{
			ptr = 0;
			try {
				buflen = in.read(buffer);
			} catch (IOException e) {
				e.printStackTrace();
			}
			if (buflen <= 0) {
				return false;
			}
		}
		return true;
	}
	private int readByte() { if (hasNextByte()) return buffer[ptr++]; else return -1;}
	private static boolean isPrintableChar(int c) { return 33 <= c && c <= 126;}
	public boolean hasNext() { while(hasNextByte() && !isPrintableChar(buffer[ptr])) ptr++; return hasNextByte();}
	public String next() {
		if (!hasNext()) throw new NoSuchElementException();
		StringBuilder sb = new StringBuilder();
		int b = readByte();
		while(isPrintableChar(b)) {
			sb.appendCodePoint(b);
			b = readByte();
		}
		return sb.toString();
	}
	public long nextLong() {
		if (!hasNext()) throw new NoSuchElementException();
		long n = 0;
		boolean minus = false;
		int b = readByte();
		if (b == '-') {
			minus = true;
			b = readByte();
		}
		if (b < '0' || '9' < b) {
			throw new NumberFormatException();
		}
		while(true){
			if ('0' <= b && b <= '9') {
				n *= 10;
				n += b - '0';
			}else if(b == -1 || !isPrintableChar(b)){
				return minus ? -n : n;
			}else{
				throw new NumberFormatException();
			}
			b = readByte();
		}
	}
	public int nextInt() {
		long nl = nextLong();
		if (nl < Integer.MIN_VALUE || nl > Integer.MAX_VALUE) throw new NumberFormatException();
		return (int) nl;
	}
	public double nextDouble() { return Double.parseDouble(next());}
}
yukicoder

結果

ソースコード
