Merge pull request #118 from YoonYn9915/main

YoonYn9915 / 1월 1주차 / 3문제

Author: YoonYn9915

YoonYn9915/dp/2025-01-04-[백준]-#15681-트리와 쿼리.py

@@ -0,0 +1,39 @@
+'''
+1. 트리 정보 입력받기. 무방향 그래프이기 때문에 간선을 잇는 두 노드에 다 저장.
+2. 부모 노드의 서브 트리의 수는 모든 자식 노드의 서브 트리 수의 합으로 분할 할 수 있다.
+    즉 분할 & 정복 가능하며 그래프를 순회하며 각 정점에서 서브트리의 수를 메모이제이션 해야 한다.
+
+'''
+
+import sys
+
+inp = sys.stdin.readline
+sys.setrecursionlimit(10 ** 6)
+
+N, R, Q = map(int, inp().split())
+
+subtree = [0] * (N + 1)
+graph = [[] for _ in range(N + 1)]
+
+for _ in range(N - 1):
+    first_node, second_node = map(int, inp().split())
+    graph[first_node].append(second_node)
+    graph[second_node].append(first_node)
+
+
+# node_num: 현재 노드 번호
+def count_subtree(graph, subtree, node_num):
+    # 방문 체크
+    subtree[node_num] = 1
+
+    for child_node in graph[node_num]:
+        if subtree[child_node] == 0:
+            count_subtree(graph, subtree, child_node)
+            subtree[node_num] += subtree[child_node]
+
+
+count_subtree(graph, subtree, R)
+
+for i in range(Q):
+    node_num = int(inp())
+    print(subtree[node_num])

YoonYn9915/dp/2025-01-04-[백준]-#1949-우수 마을.py

@@ -0,0 +1,28 @@
+import sys, collections
+
+sys.setrecursionlimit(10 ** 6)
+
+
+def dfs(cur):
+    visited[cur] = 1
+    for u in g[cur]:
+        if not visited[u]:
+            dfs(u)
+            dp[cur][1] += dp[u][0]
+            dp[cur][0] += max(dp[u][0], dp[u][1])
+
+
+n = int(sys.stdin.readline().strip())
+cost = [0] + [int(x) for x in sys.stdin.readline().split()]
+
+visited = [0 for _ in range(n + 1)]
+dp = [[0, cost[i]] * 2 for i in range(n + 1)]
+g = collections.defaultdict(list)
+
+for _ in range(n - 1):
+    v, u = map(int, sys.stdin.readline().split())
+    g[v].append(u)
+    g[u].append(v)
+
+dfs(1)
+print(max(dp[1][1], dp[1][0]))

YoonYn9915/dp/2025-01-04-[백준]-#9465-스티커.py

@@ -0,0 +1,20 @@
+t = int(input())
+
+for i in range(t):
+    col = int(input())
+    graph=[]
+    for j in range(2):
+        graph.append(list(map(int, input().split())))
+
+    if col == 1:
+        print(max(graph[0][0], graph[1][0]))
+        continue
+
+    graph[0][1] += graph[1][0]
+    graph[1][1] += graph[0][0]
+
+    for k in range(2, col):
+        graph[0][k] += max(graph[1][k-1], graph[1][k-2])
+        graph[1][k] += max(graph[0][k-1], graph[0][k-2])
+
+    print(max(graph[0][col-1], graph[1][col-1]))