[IndexedTree] baekjoon-5676

Author: Seogeurim

README.md

@@ -85,13 +85,14 @@
 | 02  |     | [Baekjoon-1927 최소 힙](./src/Heap/P1927)   |      |
 | 03  |     | [Baekjoon-11279 최대 힙](./src/Heap/P11279) |      |
 
-### Indexed Tree
+### Indexed Tree (Segment Tree)
 
 |  #  |  ☆  | Problem                                                        | Note |
 | :-: | :-: | :------------------------------------------------------------- | :--- |
 | 01  |     | [Indexed Tree Example](./src/IndexedTree/IndexedTreeTest.java) |      |
-| 02  | ⭐️ | [Baekjoon-2042 구간 합 구하기](./src/IndexedTree/P2042)        |      |
-| 03  | ⭐️ | [Baekjoon-2243 사탕상자](./src/IndexedTree/P2243)              |      |
+| 02  | ⭐️ | [Baekjoon-2042 구간 합 구하기](./src/IndexedTree/P2042)             |      |
+| 03  | ⭐️ | [Baekjoon-2243 사탕상자](./src/IndexedTree/P2243)                  |      |
+| 04  |     | [Baekjoon-5676 음주 코딩](./src/IndexedTree/P5676)               |      |
 
 ### Trie
 

src/IndexedTree/P5676/Main.java

@@ -0,0 +1,91 @@
+package IndexedTree.P5676;
+
+import java.io.*;
+import java.util.*;
+
+public class Main {
+
+    static int N, K;
+    static int[] nums;
+    final static char CH_CMD = 'C', MUL_CMD = 'P';
+
+    public static void main(String[] args) throws Exception {
+        System.setIn(new FileInputStream("src/IndexedTree/P5676/input.txt"));
+        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
+
+        String line;
+        while ((line = br.readLine()) != null) {
+            StringTokenizer st = new StringTokenizer(line);
+            N = stoi(st.nextToken());
+            K = stoi(st.nextToken());
+
+            st = new StringTokenizer(br.readLine());
+            nums = new int[N];
+            for (int i = 0; i < N; i++) nums[i] = stoi(st.nextToken());
+
+            SegmentTree tree = new SegmentTree(N, nums);
+
+            StringBuilder sb = new StringBuilder();
+            while (K-- > 0) {
+                st = new StringTokenizer(br.readLine());
+                char command = st.nextToken().charAt(0);
+                int from = stoi(st.nextToken()), to = stoi(st.nextToken());
+
+                if (command == CH_CMD) tree.changeValue(0, 0, N-1, from-1, to);
+                else if (command == MUL_CMD) {
+                    int result = tree.multiple(0, 0, N-1, from-1, to-1);
+                    if (result == 0) sb.append(0);
+                    else sb.append(result > 0 ? '+' : '-');
+                }
+            }
+            System.out.println(sb.toString());
+        }
+
+        br.close();
+    }
+
+    static int stoi(String s) { return Integer.parseInt(s); }
+}
+
+class SegmentTree {
+    int[] tree;
+    int[] arr;
+
+    public SegmentTree(int N, int[] arr) {
+        this.tree = new int[(int) Math.pow(2, (Math.ceil(Math.log(N) / Math.log(2)) + 1))];
+        this.arr = arr;
+        makeTree(0, 0, N-1);
+    }
+
+    private int setValue(int value) {
+        return Integer.compare(value, 0);
+    }
+
+    void makeTree(int idx, int start, int end) {
+        if (start == end) {
+            tree[idx] = setValue(arr[start]);
+            return;
+        }
+        makeTree(idx*2+1, start, (start+end)/2);
+        makeTree(idx*2+2, (start+end)/2+1, end);
+        tree[idx] = tree[idx*2+1] * tree[idx*2+2];
+    }
+
+    void changeValue(int idx, int start, int end, int target, int value) {
+        if (target < start || target > end) return;
+        if (start == end) {
+            tree[idx] = setValue(value);
+            return;
+        }
+        changeValue(idx*2+1, start, (start+end)/2, target, value);
+        changeValue(idx*2+2, (start+end)/2+1, end, target, value);
+        tree[idx] = tree[idx*2+1] * tree[idx*2+2];
+    }
+
+    int multiple(int idx, int start, int end, int left, int right) {
+        if (end < left || start > right) return 1;
+        if (left <= start && end <= right) return tree[idx];
+        return multiple(idx*2+1, start, (start+end)/2, left, right)
+                * multiple(idx*2+2, (start+end)/2+1, end, left, right);
+    }
+}

src/IndexedTree/P5676/README.md

@@ -0,0 +1,32 @@
+## [baekjoon-5676] 음주 코딩
+
+![image](https://user-images.githubusercontent.com/22045163/104617025-4ffa7d80-56ce-11eb-8ce1-4c44bd1354bf.png)
+![image](https://user-images.githubusercontent.com/22045163/104617067-5ab51280-56ce-11eb-93bb-62c7c117acc0.png)
+
+### 오버플로우 이슈
+
+처음에는 Segment Tree에 모든 값들을 저장하고 곱해서 풀었다. 
+그랬을 때 `틀렸습니다!`를 볼 수 있었고, 
+문제의 범위를 보니 최대의 절댓값 100, 수열의 크기는 10^5로 
+최댓값으로는 100을 10^5번 곱할 수 있다는 것인데, 거의 상상 초월로 큰 수가 나올 수 있다. 
+다음의 예시만 살펴봐도 그렇다.
+
+```
+// Input
+10 1
+100 100 100 100 100 100 100 100 100 10
+P 1 10
+// Output (Debugging)
+[-1981284352, 1410065408, 1000000000, 1000000, 10000, 1000000, 1000, 10000, 100, 100, 100, 10000, 100, 100, 10, 100, 100, 0, 0, 0, 0, 0, 0, 100, 100, 0, 0, 0, 0, 0, 0, 0]
+-1981284352
+-
+```
+
+배열에는 양수밖에 담지 않았는데, 음수의 결과가 나오는 것을 볼 수 있다.
+
+따라서 이 문제는 **값이 아닌 부호**로 풀어야 한다. 출력값으로 부호만을 요구하는 이유가 다 있다. ㅎㅎ 
+나는 1, -1, 0 으로 값을 저장하여 문제를 풀었다.
+
+![image](https://user-images.githubusercontent.com/22045163/104617857-4faeb200-56cf-11eb-8292-46db1e42d653.png)
+
+재밌다. 너무 재밌는 문제다.

src/IndexedTree/P5676/input.txt

@@ -0,0 +1,26 @@
+3 2
+0 0 0
+C 3 1
+P 3 3
+10 1
+100 100 100 100 100 100 100 100 100 10
+P 1 10
+4 6
+-2 6 0 -1
+C 1 10
+P 1 4
+C 3 7
+P 2 2
+C 4 -5
+P 1 4
+5 9
+1 5 -2 4 3
+P 1 2
+P 1 5
+C 4 -5
+P 1 5
+P 4 5
+C 3 0
+P 1 5
+C 4 -5
+C 4 -52