백준_python 9934번 완전 이진 트리 (트리, 재귀)

<문제 링크>

https://www.acmicpc.net/problem/9934

 

 

<정답 코드>

sol ) 재귀를 사용한 완전 이진트리 구성

def makeTree(arr, level):
    mid = len(arr) // 2
    tree[level].append(arr[mid])
    
    # arr의 원소가 1개이면 마지막 레벨이므로 return
    if len(arr) == 1:
        return
    
    # level+1 한 후 왼쪽, 오른쪽 자식 노드 처리
    makeTree(arr[:mid], level + 1)
    makeTree(arr[mid + 1:], level + 1)
 
def main():
    k = int(input())
    bin_tree = list(map(int, input().split()))
    
    # 각 level의 원소 모을 배열 생성
    global tree
    tree = [[] for _ in range(k)]
    
    makeTree(bin_tree, 0)
    
    # 결과 출력
    for i in range(k):
        print(*tree[i])
 
if __name__ == '__main__':
    main()

 

 

완전 이진트리에 대한 개념은 있었지만, 재귀를 사용해 문제 푸는 아이디어를 떠올리지 못했다.