반응형 알고리즘/트리19 [백준][Python] 트리의 높이와 너비 2250번: 트리의 높이와 너비 첫째 줄에 노드의 개수를 나타내는 정수 N(1 ≤ N ≤ 10,000)이 주어진다. 다음 N개의 줄에는 각 줄마다 노드 번호와 해당 노드의 왼쪽 자식 노드와 오른쪽 자식 노드의 번호가 순서대로 주어진다. www.acmicpc.net 코드 def inorder(node,level): global dist if n_list[node][0] != -1: inorder(n_list[node][0],level+1) distance[level].append(dist) dist += 1 if n_list[node][1] != -1: inorder(n_list[node][1],level+1) n = int(input()) n_list = [[0,0] for _ in range(n+1)].. 2022. 11. 17. [백준][Python] 2533번 사회망 서비스(SNS) 2533번: 사회망 서비스(SNS) 첫 번째 줄에는 친구 관계 트리의 정점 개수 N이 주어진다. 단, 2 ≤ N ≤ 1,000,000이며, 각 정점은 1부터 N까지 일련번호로 표현된다. 두 번째 줄부터 N-1개의 줄에는 각 줄마다 친구 관계 트리의 에 www.acmicpc.net 코드 import sys sys.setrecursionlimit(10**9) def dfs(node): visit[node] = 1 for i in graph[node]: if visit[i] == 0: dfs(i) dp[node][0] += min(dp[i][0],dp[i][1]) dp[node][1] += dp[i][0] n = int(input()) graph = [[] for _ in range(n+1)] dp = [[1.. 2022. 10. 17. [백준][Python] 15681번 트리와 쿼리 15681번: 트리와 쿼리 트리의 정점의 수 N과 루트의 번호 R, 쿼리의 수 Q가 주어진다. (2 ≤ N ≤ 105, 1 ≤ R ≤ N, 1 ≤ Q ≤ 105) 이어 N-1줄에 걸쳐, U V의 형태로 트리에 속한 간선의 정보가 주어진다. (1 ≤ U, V ≤ N, U ≠ V) www.acmicpc.net 코드 import sys sys.setrecursionlimit(10**9) def tree(node,par): for i in graph[node]: if i != par: child[node].append(i) parent[i] = node tree(i,node) def count(node): size[node] = 1 for i in child[node]: count(i) size[node] +=.. 2022. 10. 14. [백준][Python] 11437번 LCA 11437번: LCA 첫째 줄에 노드의 개수 N이 주어지고, 다음 N-1개 줄에는 트리 상에서 연결된 두 정점이 주어진다. 그 다음 줄에는 가장 가까운 공통 조상을 알고싶은 쌍의 개수 M이 주어지고, 다음 M개 줄에는 정 www.acmicpc.net 코드 import sys sys.setrecursionlimit(100000) def dfs(num,dep): visit[num] = 1 d[num] = dep for i in tree[num]: if visit[i] == 1: continue par[i] = num dfs(i,dep+1) def parent(a,b): while d[a] != d[b]: if d[a] < d[b]: b = par[b] else: a = par[a] while a != b: .. 2022. 10. 13. 이전 1 2 3 4 5 다음 반응형
