[ BOJ ][JAVA][12757] 전설의 JBNU

https://www.acmicpc.net/problem/12757

12757번: 전설의 JBNU

시간 제한	메모리 제한	제출	정답	맞은 사람	정답 비율
2 초	128 MB	481	135	107	33.127%

문제

전설의 프로그래머 윤준하는 독자적인 데이터베이스 시스템 JBNU(Jeong Bo Neoh Um)를 만들었다.

준하가 생각한 데이터베이스의 기본 골자는 데이터에 접근하기 위한 Key와 그 데이터를 나타내는 Value로 구성되어 있다. 사용자는 Key를 알고 있어야만 원하는 데이터에 접근할 수 있다.

하지만 준하는 건망증이 심해 Key를 매번 잊어버리기 일쑤였다. 따라서 준하는 JBNU를 개조하여 잘못된 Key를 입력하더라도 그 잘못된 Key와 제일 근접한 Key를 찾아주는 메커니즘을 도입하였다.

Key와 Value는 항상 정수로 되어있다. 가장 근접한 Key란 두 수의 차이가 가장 작은 Key를 의미한다. 또한, 정보의 정확성을 위해 두 수의 차이가 K보다 큰 경우는 Key로 인정하지 않기로 하였다.

프로젝트 베끼기의 달인 승균이는 데이터베이스 시간에 JBNU를 모방하기로 했다. 그러나 준하는 전설이기 때문에 그가 만든 프로그램은 찾을 방법이 없었고, 하는 수 없이 같은 조원인 당신에게 맡기려고 한다.

JBNU의 초기 데이터 상태가 주어질 때, 데이터 추가, 수정 및 검색을 지원하는 프로그램을 작성해보자.

입력

첫 줄에는 초기 데이터의 개수인 N(1≤N≤100,000) 과 명령 횟수인 M(1≤M≤100,000), 가장 근접한 Key까지의 거리의 제한인 K(1≤K≤10,000)가 주어진다.

입력의 둘째 줄부터 N개의 줄에는 초기 데이터인 Key와 Value 값이 주어진다. 모든 Key와 Value는 1,000,000,000 이하의 음이 아닌 정수이다. 같은 Key를 갖는 데이터는 없다.

다음 M개의 줄에는 아래와 같은 명령이 주어진다.

• 1 Key Value : 해당 Key와 Value를 가진 데이터를 추가한다. Key가 이미 존재하는 입력은 주어지지 않는다.
• 2 Key Value : 해당 Key로 검색된 데이터를 Value로 변경한다. 조건을 만족하는 유일한 Key가 없는 경우 무시한다.
• 3 Key : 해당 Key로 검색된 데이터를 출력한다. 조건을 만족하는 Key가 없는 경우 -1을 출력한다. 만약 해당하는 Key가 두 개 이상 존재한다면 ?를 출력한다. 모든 출력은 개행을 포함해야 한다.

출력

각 줄에 걸쳐 3번 명령에 대한 결과를 출력한다.

예제 입력 1

5 7 5
1 10
5 20
9 30
15 40
50 50
3 2
2 0 35
3 2
3 7
3 100
1 97 123
3 100

예제 출력 1

10
35
?
-1
123

접근

TreeMap을 이용해 모든 key를 순차적으로 정렬하게 한다.

그 후 TreeMap의 내장함수를 통해 lowerbound / upperbound를 구현한다.

ceilingKey(key) : treeMap에서 key보다 크거나 같은 값 중에 최소 키 값
floorKey(key) : treeMap에서 key보다 작거나 같은 값 중에 최대 키 값

코드

/*
    전설의 JBNU
    TreeMap
     - ceilingKey(key) : treeMap에서 key보다 크거나 같은 값 중에 최소 키 값
     - floorKey(key) : treeMap에서 key보다 작거나 같은 값 중에 최대 키 값
*/
import java.io.*;
import java.util.*;
public class p12757 {
    static int n, m, k;
    static TreeMap<Integer, Integer> tm;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        n = Integer.parseInt(st.nextToken());
        m = Integer.parseInt(st.nextToken());
        k = Integer.parseInt(st.nextToken());

        // 초기 데이터
        tm = new TreeMap<>();
        tm.put(-2000000000, -1);
        tm.put(2000000000, -1);
        for (int i = 0; i < n; i++) {
            st = new StringTokenizer(br.readLine());
            int key = Integer.parseInt(st.nextToken());
            int value = Integer.parseInt(st.nextToken());
            tm.put(key, value);
        }

        for (int i = 0; i < m; i++) {
            st = new StringTokenizer(br.readLine());
            int command = Integer.parseInt(st.nextToken());

            if (command == 1) { // 데이터 추가
                int key = Integer.parseInt(st.nextToken());
                int value = Integer.parseInt(st.nextToken());
                tm.put(key, value);
            } else if (command == 2) { // 데이터 수정
                int key = Integer.parseInt(st.nextToken());
                int value = Integer.parseInt(st.nextToken());
                update(key, value);
            } else { // 데이터 출력
                int key = Integer.parseInt(st.nextToken());
                print(key);
            }
        }
    }

    static void update(int key, int value) {
        // ceilingKey(key) : treeMap에서 key보다 크거나 같은 값 중에 최소 키 값
        // floorKey(key) : treeMap에서 key보다 작거나 같은 값 중에 최대 키 값
        int rightKey = tm.ceilingKey(key);
        int leftKey = tm.floorKey(key);

        if (tm.get(rightKey) == tm.get(leftKey)) {
            tm.put(key, value);
        } else if (rightKey - key < key - leftKey && k >= rightKey - key) {
            tm.put(rightKey, value);
        } else if (rightKey - key > key - leftKey && k >= key - leftKey) {
            tm.put(leftKey, value);
        }
    }

    static void print(int key) {
        int rightKey = tm.ceilingKey(key);
        int leftKey = tm.floorKey(key);

        if (tm.get(rightKey) == tm.get(leftKey)) {
            System.out.println(tm.get(key));
        } else if (rightKey - key == key - leftKey && k >= rightKey - key) {
            System.out.println("?");
        } else if (rightKey - key < key - leftKey && k >= rightKey - key) {
            System.out.println(tm.get(rightKey));
        } else if (rightKey - key > key - leftKey && k >= key - leftKey) {
            System.out.println(tm.get(leftKey));
        } else {
            System.out.println("-1");
        }
    }
}