76-tgyuuAn

tgyuuAn/README.md

@@ -82,4 +82,5 @@
 | 73차시 | 2024.08.26 |  BFS  | <a href="https://www.acmicpc.net/problem/14324">Rain (Small)</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/239
 | 74차시 | 2024.08.30 |  BFS  | <a href="https://www.acmicpc.net/problem/11967">불 켜기</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/242
 | 75차시 | 2024.09.02 |  DP  | <a href="https://school.programmers.co.kr/learn/courses/30/lessons/258705">산 모양 타일링</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/243
+| 76차시 | 2024.09.06 |  DFS + 트리  | <a href="https://school.programmers.co.kr/learn/courses/30/lessons/150367">표현 가능한 이진트리</a> | https://github.com/AlgoLeadMe/AlgoLeadMe-1/pull/246
 ---

tgyuuAn/트리/표현 가능한 이진트리.py

@@ -0,0 +1,32 @@
+def dfs(b, i, depth):
+    if depth == 0:  	# 리프 노드에 도달했다면
+        return True 	# 포화이진트리
+
+    # 부모노드가 '0' 일때
+    # 왼쪽 자식 노드가 '1' 이거나 오른쪽 자식 노드가 '1' 이라면 포화 이진트리가 될 수 없음
+    elif b[i] == '0':   
+        if b[i - depth] == '1' or b[i + depth] == '1': return False
+
+    # 왼쪽 서브 트리 탐색
+    left = dfs(b, i - depth, depth // 2)
+    # 오른쪽 서브 트리 탐색
+    right = dfs(b, i + depth, depth // 2)
+    return left and right
+
+
+def solution(numbers):
+    answer = []
+    for num in numbers:				# num = 42
+        b = bin(num)[2:]  			# b = 101010 / len(b) = 6
+        nodes = bin(len(b) + 1)[2:] 	# nodes = 7 = 111
+
+        # 포화이진트리가 아닌 경우 더미노드(0추가)
+        if '1' in nodes[1:]:       
+            dummies = (1 << len(nodes)) - int(nodes, 2)
+            b = '0' * dummies + b
+
+        # 이미 포화이진트리일 경우
+        result = dfs(b, len(b)//2, (len(b)+1)//4)
+        answer.append(1 if result else 0)
+
+    return answer