Merge pull request #162 from AlgoLeadMe/45-tgyuuAn

45-tgyuuAn

tgyuuAn/README.md

@@ -39,11 +39,12 @@
 | 35차시 | 2023.01.25 |  이분 탐색  | <a href="https://www.acmicpc.net/problem/2110">공유기 설치</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/120
 | 36차시 | 2023.02.04 |  BFS  | <a href="https://www.acmicpc.net/problem/4991">로봇 청소기</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/126
 | 37차시 | 2023.02.04 |  BFS  | <a href="https://www.acmicpc.net/problem/1039">교환</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/131
-| 38차시 | 2023.02.15 |  DP  | <a href="https://www.acmicpc.net/problem/2749">피보나치 수 3</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/137
+| 38차시 | 2023.02.15 |  DP  | <a href="https://www.acmicpc.net/problem/2749">피보나치 수 3</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/138
 | 39차시 | 2023.02.18 |  DP  | <a href="https://www.acmicpc.net/problem/7579">앱</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/139
 | 40차시 | 2023.02.21 |  DP  | <a href="https://www.acmicpc.net/problem/31413">입대</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/142
 | 41차시 | 2023.03.04 |  DP  | <a href="https://www.acmicpc.net/problem/2240">자두나무</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/148
 | 42차시 | 2023.03.07 |  DFS  | <a href="https://www.acmicpc.net/problem/2239">스도쿠</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/152
 | 43차시 | 2024.03.10 |  이분 탐색  | <a href="https://school.programmers.co.kr/learn/courses/30/lessons/64062">징검다리 건너기</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/157
 | 44차시 | 2023.03.13 |  트라이  | <a href="https://www.acmicpc.net/problem/14725">개미굴</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/159
+| 45차시 | 2023.03.16 |  트라이  | <a href="https://www.acmicpc.net/problem/31413">트라이</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/162
 ---

tgyuuAn/트라이/디스크 트리.py

@@ -0,0 +1,48 @@
+import sys
+
+def input(): return sys.stdin.readline().rstrip()
+
+N = int(input())
+
+class Node():
+    def __init__(self, key):
+        self.key = key
+        self.children = {}
+
+class Tries():
+    def __init__(self):
+        self.head = Node(None)
+
+    def insert(self, path):
+        now = self.head
+        directories = path.split("\\")
+
+        for directory in directories:
+            if directory not in now.children:
+                now.children[directory] = Node(directory)
+
+            now = now.children[directory]
+
+    def dfs(self, now : Node, depth):
+        if now.key is not None:
+            for _ in range(depth):
+                print(" ", end="")
+            print(now.key)
+
+        now_children = list(now.children.keys())
+        now_children.sort()
+
+        for child in now_children:
+            self.dfs(now.children[child], depth+1)
+
+    def display_all(self):
+        now = self.head
+        self.dfs(now, -1)
+
+tries = Tries()
+
+for _ in range(N):
+    path = input()
+    tries.insert(path)
+
+tries.display_all()