Merge pull request #730 from baekhangyeol/main

[백한결] 116차 라이브 코테 제출

Author: github-actions[bot]

live11/test116/문제1/백한결.py

@@ -0,0 +1,47 @@
+import sys
+sys.setrecursionlimit(10**6)
+
+def main():
+    input = sys.stdin.readline
+    T = int(input())
+
+    for _ in range(T):
+        N = int(input())
+        parent = [0] * (N+1)
+        depth = [0] * (N+1)
+        tree = [[] for _ in range(N+1)]
+        isChild = [False] * (N+1)
+        for _ in range(N-1):
+            a, b = map(int, input().split())
+            parent[b] = a
+            tree[a].append(b)
+            isChild[b] = True
+
+        root = 0
+        for i in range(1, N+1):
+            if not isChild[i]:
+                root = i
+                break
+
+        def dfs(node, d):
+            depth[node] = d
+            for child in tree[node]:
+                dfs(child, d + 1)
+
+        dfs(root, 0)
+
+        a, b = map(int, input().split())
+
+        while depth[a] > depth[b]:
+            a = parent[a]
+        while depth[a] < depth[b]:
+            b = parent[b]
+
+        while a != b:
+            a = parent[a]
+            b = parent[b]
+
+        print(a)
+
+if __name__ == '__main__':
+    main()

live11/test116/문제2/백한결.py

@@ -0,0 +1,42 @@
+import sys
+from collections import defaultdict, deque
+
+
+def main():
+    input = sys.stdin.readline
+    n, T = map(int, input().split())
+
+    stones = set()
+    yMap = defaultdict(list)
+
+    for _ in range(n):
+        x, y = map(int, input().split())
+        stones.add((x, y))
+        yMap[y].append(x)
+
+    for y in yMap:
+        yMap[y].sort()
+
+    visited = set()
+    queue = deque()
+    queue.append(((0, 0), 0)) # 위치 (x, y), 이동 횟수
+
+    while queue:
+        (x, y), count = queue.popleft()
+
+        if y == T:
+            print(count)
+            return
+
+        for ny in range(y-2, y+3):
+            if ny < 0 or ny > T:
+                continue
+            for nx in range(x-2, x+3):
+                if (nx, ny) in stones and (nx, ny) not in visited:
+                    visited.add((nx, ny))
+                    queue.append(((nx, ny), count + 1))
+
+    print(-1)
+
+if __name__ == '__main__':
+    main()