[DP] baekjoon-2629

Author: Seogeurim

README.md

@@ -401,6 +401,7 @@
 | 34  |     | [Baekjoon-12852 1로 만들기 2](./src/DP/P12852)              |        |
 | 35  |     | [Baekjoon-1149 RGB거리](./src/DP/P1149)                    |        |
 | 36  |     | [Programmers N으로 표현](./src/DP/prg42895)                 |        |
+| 37  | ⭐️ | [Baekjoon-2629 양팔저울](./src/DP/P2629)                    |        |
 
 ## Time Complexity
 

src/DP/P2629/Main.java

@@ -0,0 +1,43 @@
+package DP.P2629;
+
+import java.io.*;
+import java.util.*;
+
+public class Main {
+
+    static int N, M;
+    static int[] chu;
+    static boolean[][] check;
+
+    public static void main(String[] args) throws Exception {
+        System.setIn(new FileInputStream("src/DP/P2629/input.txt"));
+        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
+
+        N = Integer.parseInt(br.readLine());
+        chu = new int[N];
+        check = new boolean[N+1][40001];
+
+        StringTokenizer st = new StringTokenizer(br.readLine());
+        for (int i = 0; i < N; i++) chu[i] = Integer.parseInt(st.nextToken());
+
+        calc(0, 0);
+
+        M = Integer.parseInt(br.readLine());
+        st = new StringTokenizer(br.readLine());
+        while (M-- > 0) {
+            if (check[N][Integer.parseInt(st.nextToken())]) System.out.print("Y ");
+            else System.out.print("N ");
+        }
+        System.out.println();
+    }
+
+    static void calc(int i, int weight) {
+        if (i > N || weight > 40000 || check[i][weight]) return;
+        check[i][weight] = true;
+
+        if (i == N) return;
+        calc(i+1, weight);
+        calc(i+1, weight + chu[i]);
+        calc(i+1, Math.abs(weight - chu[i]));
+    }
+}

src/DP/P2629/README.md

@@ -0,0 +1,21 @@
+## [baekjoon-2629] 양팔저울
+
+![image](https://user-images.githubusercontent.com/22045163/113178190-4f16c800-9289-11eb-84be-98309288429e.png)
+![image](https://user-images.githubusercontent.com/22045163/113178261-5dfd7a80-9289-11eb-861e-0c4322f5d5c2.png)
+
+### 풀이 과정
+
+시간 제한, 메모리 제한을 따졌을 때 DP로 풀어야 한다. (처음에 조합으로 모든 경우의 수를 Set에 넣고 계산했다가 메모리 초과가 발생하였다.)
+
+그럼 한 `시점`에서 따질 수 있는 경우의 수는 **다음의 세 가지**이며, 이 때의 `시점`은 **현재 몇 번째 추까지 계산했는지**이다.
+
+```text
+현 시점 : i번째 추를 계산 중
+1. i번째 추를 사용하지 않음
+2. i번째 추를 구슬 반대쪽에 올림
+3. i번째 추를 구슬쪽에 올림
+```
+
+위 경우의 수에 대하여 메모이제이션을 해나가면서 재귀를 돌린다. 시간초과가 발생하지 않도록 **종료조건 및 방문 여부**를 신경써줘야 하는데, `현 시점에서 추를 사용했거나 / 안 사용했거나`에 따라 방문 여부의 `차원`이 달라지므로 방문 배열을 2차원 배열로 선언해야 한다.
+
+![image](https://user-images.githubusercontent.com/22045163/113178358-753c6800-9289-11eb-8586-a0e08149aff4.png)

src/DP/P2629/input.txt

@@ -0,0 +1,4 @@
+4
+2 3 3 3
+3
+1 4 10