세그먼트트리 / 인덱스를 출력하는거라 어려웠음

Author: SoTaeHo

14428번 - 수열과 쿼리 16/Main.java

@@ -0,0 +1,100 @@
+
+/* ************************************************************************** */
+/*                                                                            */
+/*                                                      :::    :::    :::     */
+/*   Problem Number: 14428                             :+:    :+:      :+:    */
+/*                                                    +:+    +:+        +:+   */
+/*   By: thxogh1 <boj.kr/u/thxogh1>                  +#+    +#+          +#+  */
+/*                                                  +#+      +#+        +#+   */
+/*   https://boj.kr/14428                          #+#        #+#      #+#    */
+/*   Solved: 2025/05/06 15:35:34 by thxogh1       ###          ###   ##.kr    */
+/*                                                                            */
+/* ************************************************************************** */
+import java.io.BufferedReader;
+import java.io.InputStreamReader;
+import java.util.StringTokenizer;
+
+public class Main {
+
+    static void init(int[] arr, int[] tree, int node, int s, int e) {
+        if (s == e) {
+            tree[node] = s;
+        } else {
+            int m = (s + e) / 2;
+            init(arr, tree, node * 2, s, m);
+            init(arr, tree, node * 2 + 1, m + 1, e);
+            if (arr[tree[node * 2]] <= arr[tree[node * 2 + 1]]) {
+                tree[node] = tree[node * 2];
+            } else {
+                tree[node] = tree[node * 2 + 1];
+            }
+        }
+    }
+
+    static void update(int[] arr, int[] tree, int node, int s, int e, int idx, int val) {
+        if (e < idx || idx < s) {
+            return;
+        }
+        if (s == e && idx == s) {
+            arr[tree[node]] = val;
+            return;
+        }
+        int m = (s + e) / 2;
+        update(arr, tree, node * 2, s, m, idx, val);
+        update(arr, tree, node * 2 + 1, m + 1, e, idx, val);
+        if (arr[tree[node * 2]] <= arr[tree[node * 2 + 1]]) {
+            tree[node] = tree[node * 2];
+        } else {
+            tree[node] = tree[node * 2 + 1];
+        }
+    }
+
+    static int query(int[] arr, int[] tree, int node, int s, int e, int l, int r) {
+        if (e < l || r < s) {
+            return -1;
+        }
+        if (l <= s && e <= r) {
+            return tree[node];
+        }
+        int m = (s + e) / 2;
+        int lmin = query(arr, tree, node * 2, s, m, l, r);
+        int rmin = query(arr, tree, node * 2 + 1, m + 1, e, l, r);
+        if (lmin == -1) {
+            return rmin;
+        } else if (rmin == -1) {
+            return lmin;
+        } else if (arr[lmin] <= arr[rmin]) {
+            return lmin;
+        } else {
+            return rmin;
+        }
+    }
+
+    public static void main(String[] args) throws Exception {
+        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
+        StringTokenizer st;
+
+        int n = Integer.parseInt(br.readLine());
+        int[] arr = new int[n];
+        st = new StringTokenizer(br.readLine());
+        for (int i = 0; i < n; i++) {
+            arr[i] = Integer.parseInt(st.nextToken());
+        }
+        int m = Integer.parseInt(br.readLine());
+        int[] tree = new int[4 * n];
+        init(arr, tree, 1, 0, n - 1);
+        for (int i = 0; i < m; i++) {
+            st = new StringTokenizer(br.readLine());
+            int type = Integer.parseInt(st.nextToken());
+            if (type == 1) {
+                int idx = Integer.parseInt(st.nextToken());
+                int val = Integer.parseInt(st.nextToken());
+                update(arr, tree, 1, 0, n - 1, idx - 1, val);
+            } else {
+                int s = Integer.parseInt(st.nextToken());
+                int e = Integer.parseInt(st.nextToken());
+                System.out.println(query(arr, tree, 1, 0, n - 1, s - 1, e - 1) + 1);
+            }
+        }
+    }
+}

14428번 - 수열과 쿼리 16/README.md

@@ -0,0 +1,38 @@
+# 14428번: 수열과 쿼리 16 - <img src="https://static.solved.ac/tier_small/15.svg" style="height:20px" /> Gold I
+
+<!-- performance -->
+
+<!-- 문제 제출 후 깃허브에 푸시를 했을 때 제출한 코드의 성능이 입력될 공간입니다.-->
+
+<!-- end -->
+
+## 문제
+
+[문제 링크](https://boj.kr/14428)
+
+<p>길이가 N인 수열 A<sub>1</sub>, A<sub>2</sub>, ..., A<sub>N</sub>이 주어진다. 이때, 다음 쿼리를 수행하는 프로그램을 작성하시오.</p>
+
+<ul>
+<li><code>1 i v</code> : A<sub>i</sub>를 v로 바꾼다. (1 ≤ i ≤ N, 1 ≤ v ≤ 10<sup>9</sup>)</li>
+<li><code>2 i j</code> : A<sub>i</sub>, A<sub>i+1</sub>, ..., A<sub>j</sub>에서 크기가 가장 작은 값의 인덱스를 출력한다. 그러한 값이 여러개인 경우에는 인덱스가 작은 것을 출력한다. (1 ≤ i ≤ j <span style="display: none;"> </span>≤ N, 1 ≤ v ≤ 10<sup>9</sup>)</li>
+</ul>
+
+<p>수열의 인덱스는 1부터 시작한다.<span style="display: none;"> </span></p>
+
+## 입력
+
+<p>첫째 줄에 수열의 크기 N이 주어진다. (1 ≤ N ≤ 100,000)</p>
+
+<p>둘째 줄에는 A<sub>1</sub>, A<sub>2</sub>, ..., A<sub>N</sub>이 주어진다. (1 ≤ A<sub>i</sub> ≤ 10<sup>9</sup>)</p>
+
+<p>셋째 줄에는 쿼리의 개수 M이 주어진다. (1 ≤ M ≤ 100,000)</p>
+
+<p>넷째 줄부터 M개의 줄에는 쿼리가 주어진다.</p>
+
+## 출력
+
+<p>2번 쿼리에 대해서 정답을 한 줄에 하나씩 순서대로 출력한다.</p>
+
+## 소스코드
+
+[소스코드 보기](Main.java)