[Java] 백준 10868 최솟값 - 세그먼트 트리(2)

백준 10868 최솟값
문제

N(1 ≤ N ≤ 100,000)개의 정수들이 있을 때, a번째 정수부터 b번째 정수까지 중에서 제일 작은 정수를 찾는 것은 어려운 일이 아니다. 하지만 이와 같은 a, b의 쌍이 M(1 ≤ M ≤ 100,000)개 주어졌을 때는 어려운 문제가 된다. 이 문제를 해결해 보자.

여기서 a번째라는 것은 입력되는 순서로 a번째라는 이야기이다. 예를 들어 a=1, b=3이라면 입력된 순서대로 1번, 2번, 3번 정수 중에서 최솟값을 찾아야 한다. 각각의 정수들은 1이상 1,000,000,000이하의 값을 갖는다.

입력

첫째 줄에 N, M이 주어진다. 다음 N개의 줄에는 N개의 정수가 주어진다. 다음 M개의 줄에는 a, b의 쌍이 주어진다.

출력

M개의 줄에 입력받은 순서대로 각 a, b에 대한 답을 출력한다.

---

제출

import java.io.*;
import java.util.*;
public class Main {
	static long[] tree;
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		int N = Integer.parseInt(st.nextToken());
		int M = Integer.parseInt(st.nextToken());
		int treeHeight = 0;
		int lenght = N;
		while (lenght != 0) {
			lenght /= 2;
			treeHeight++;
		}
		int treeSize = (int)Math.pow(2,  treeHeight + 1);
		int leftNodeStartIndex = treeSize / 2 - 1;
		tree = new long[treeSize + 1];
		for (int i = 0; i < tree.length; i++) {
			tree[i] = Integer.MAX_VALUE;
		}
		for (int i = leftNodeStartIndex + 1; i <= leftNodeStartIndex + N; i++) {
			tree[i] = Long.parseLong(br.readLine());
		}
		setTree(treeSize - 1);
		for (int i =0; i < M; i++) {
			st = new StringTokenizer(br.readLine());
			int s = Integer.parseInt(st.nextToken());
			int e = Integer.parseInt(st.nextToken());
			s = s + leftNodeStartIndex;
			e = e + leftNodeStartIndex;
			System.out.println(getMin(s, e));
		}
		br.close();
	}
	private static long getMin(int s, int e) {
		long Min = Long.MAX_VALUE;
		while (s <= e) {
			if (s % 2 == 1) {
				Min = Math.min(Min,  tree[s]);
				s++;
			}
			s = s / 2;
			if (e % 2 == 0) {
				Min = Math.min(Min,  tree[e]);
				e--;
			}
			e = e / 2;
		}
		return Min;
	}
	private static void setTree(int i) {
		while (i != 1) {
			if (tree[i / 2] > tree[i])
				tree[i / 2] = tree[i];
			i--;
		}
	}
}

예제

10 4
75
30
100
38
50
51
52
20
81
5
1 10
3 5
6 9
8 10

결과

백준 10868 최솟값

 

10868번: 최솟값