Merge pull request #116 from AlgorithmStudy-Allumbus/minjeong

feat: 1월 1주차 발제 문제 내용

Author: Mingguriguri

_WeeklyChallenges/W06-[Tree-DP]/Assignment_BOJ_1949_우수마을.py

@@ -0,0 +1,44 @@
+'''
+BOJ #1949. 우수마을 (골드2)
+https://www.acmicpc.net/problem/1949
+유형: Dynamic Programming(DP), Tree DP, tree
+'''
+import sys
+
+input = sys.stdin.readline
+sys.setrecursionlimit(10 ** 6)  # 재귀 깊이 설정 (10만)
+
+# 1. 입력 받기
+N = int(input())  # 마을 개수
+people = [0] + list(map(int, input().split()))  # 각 마을 주민 수 (인덱스 맞추기 위해 0 추가)
+
+# 2. 마을 연결 (트리 구성)
+town = [[] for _ in range(N + 1)]
+
+for _ in range(N - 1):
+    u, v = map(int, input().split())
+    town[u].append(v)
+    town[v].append(u)
+
+# 3. DP 테이블 초기화 및 방문 리스트
+dp = [[0, 0] for _ in range(N + 1)]  # dp[i][0]: 우수 마을 X, dp[i][1]: 우수 마을 O
+visited = [False] * (N + 1)
+
+
+# 4. DFS를 통한 서브트리 탐색 및 DP 계산
+def dfs(current):
+    visited[current] = True  # 현재 노드 방문 처리
+    dp[current][1] += people[current]  # 현재 마을을 우수 마을로 선정한 경우 (자기 자신 포함)
+
+    for child in town[current]:
+        if not visited[child]:
+            dfs(child)  # 자식 노드로 DFS 진행
+            dp[current][1] += dp[child][0]  # 자식 노드가 우수 마을이 아닌 경우 주민 수 더하기
+            dp[current][0] += max(dp[child][0], dp[child][1])  # 자식 노드에서 우수 마을이거나 아닌 경우 중 최대값 선택
+
+
+# 5. DFS 호출 (루트 노드 1부터 시작)
+dfs(1)
+
+# 6. 결과 출력
+print(max(dp[1][0], dp[1][1]))

_WeeklyChallenges/W06-[Tree-DP]/README.md

@@ -0,0 +1,13 @@
+## 🚀1월 1주차 (12/30) 스터디 발제 주제: Tree Dynamic Programming
+> 발제자: 김민정 (@Mingguriguri)
+
+### 🗂️ 스터디 자료
+- PDF: [바로가기](https://github.com/AlgorithmStudy-Allumbus/codingtest_algorithm_study/blob/main/_WeeklyChallenges/W02-%5BDP%5D/Study_BOJ_1125.pdf)
+
+### 📖 문제
+- [백준 #15681. 트리와 쿼리](https://www.acmicpc.net/problem/15681): Tree DP / 골드5
+- 정답 코드: [Study_BOJ_15681_트리와쿼리.py](https://github.com/AlgorithmStudy-Allumbus/codingtest_algorithm_study/blob/main/_WeeklyChallenges/W02-%5BDP%5D/Study_BOJ_12865_%ED%8F%89%EB%B2%94%ED%95%9C%EB%B0%B0%EB%82%AD.py#L1-L30)
+
+### 💻 과제
+- [백준 #1949. 우수마을](https://www.acmicpc.net/problem/1949): Tree DP / 골드2
+- 정답 코드: [Assignment_BOJ_1949_우수마을.py](https://github.com/AlgorithmStudy-Allumbus/codingtest_algorithm_study/blob/main/_WeeklyChallenges/W02-%5BDP%5D/Assignment_BOJ_17845_%EC%88%98%EA%B0%95%EA%B3%BC%EB%AA%A9.py#L1-L20)

_WeeklyChallenges/W06-[Tree-DP]/Study_BOJ_15681.pdf

@@ -0,0 +1,37 @@
+'''
+BOJ #15681. 트리와 쿼리(골드5)
+https://www.acmicpc.net/problem/15681
+유형: Dynamic Programming(DP), Tree DP, tree
+'''
+import sys
+input = sys.stdin.readline
+sys.setrecursionlimit(10**6)  # 재귀 깊이 제한 설정(10만)
+
+# 1. 입력 처리
+N, R, Q = map(int, input().split())  # 정점 수, 루트 번호, 쿼리 수
+graph = [[] for _ in range(N + 1)]  # 트리 그래프
+
+# 2. 트리 정보 입력
+for _ in range(N - 1):
+    u, v = map(int, input().split())
+    graph[u].append(v)
+    graph[v].append(u)
+
+# 3. 서브트리 크기 기록 배열
+size = [0] * (N + 1)
+
+# 4. DFS를 이용한 서브트리 크기 계산
+def countSubtreeNodes(current):
+    size[current] = 1  # 자기 자신 포함
+    for node in graph[current]:
+        if size[node] == 0:  # 아직 방문하지 않은 경우
+            countSubtreeNodes(node)
+            size[current] += size[node]  # 자식 서브트리 크기 추가
+
+# 5. 루트에서 시작해 서브트리 크기 계산
+countSubtreeNodes(R)
+
+# 6. 쿼리 처리
+for _ in range(Q):
+    U = int(input())
+    print(size[U])