Merge pull request #78 from AlgoLeadMe/21-seongwon030

21-seongwon030
AlgoLeadMe · Aug 4, 2024 · cadb852 · cadb852
2 parents a59bee6 + 5c078f2
commit cadb852
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/seongwon030/dfs/트리.py b/seongwon030/dfs/트리.py
@@ -0,0 +1,21 @@
+import sys
+input = sys.stdin.readline
+
+n = int(input())
+k = list(map(int,input().split()))
+del_node = int(input())
+
+def dfs(node):
+  k[node] = -2
+  for i in range(n):
+    if node == k[i]:
+      dfs(i)
+
+dfs(del_node)
+
+cnt = 0
+for j in range(n):
+  if k[j] != -2 and j not in k:
+    cnt+=1
+
+print(cnt)