🐜 Study: 도넛과 막대 그래프

도넛과 막대 그래프.py

@@ -0,0 +1,20 @@
+from collections import defaultdict, deque
+
+def solution(edges):
+    answer = [0] * 4
+    graph = defaultdict(lambda: [[], []])
+    for e in edges:
+        graph[e[0]][1].append(e[1])
+        graph[e[1]][0].append(e[0])
+    answer[0] = next(g for g in graph
+        if not graph[g][0] and len(graph[g][1]) > 1)
+    for g in graph[answer[0]][1]:
+        sin, dou, s, q = False, False, {g}, deque([g])
+        while q:
+            cur = q.popleft()
+            for i in graph[cur][1]:
+                if i not in s: s.add(i); q.append(i)
+                elif not sin: sin = True
+                else: dou = True
+        answer[sin + (not (sin ^ dou)) * 2] += 1
+    return answer