Tiny Bunny
본문 바로가기
Algorithm/Python

[SWEA #12915] 이진탐색

by nowag 2024. 2. 20.

문제

1부터 N까지의 자연수를 이진 탐색 트리에 저장하려고 한다.
이진 탐색 트리는 어떤 경우에도 저장된 값이 왼쪽 서브트리의 루트 <현재 노드 <오른쪽 서브 트리의 루트인 규칙을 만족한다.
추가나 삭제가 없는 경우에는, 완전 이진 트리가 되도록 만들면 효율적인 이진 탐색 트리를 만들수 있다.
다음은 1부터 6까지의 숫자를 완전 이진 트리 형태인 이진 탐색 트리에 저장한 경우이다.

 

완전 이진 트리의 노드 번호는 루트를 1번으로 하고 아래로 내려가면서 왼쪽에서 오른쪽 순으로 증가한다.
N이 주어졌을 때 완전 이진 트리로 만든 이진 탐색 트리의 루트에 저장된 값과, N/2번 노드(N이 홀수인 경우 소수점 버림)에 저장된 값을 출력하는 프로그램을 만드시오.

입력
첫 줄에 테스트케이스의 수 T가 주어진다. 1<=T<=50
다음 줄부터 테스트 케이스의 별로 N이 주어진다. 1<=N<=1000

출력
각 줄마다 "#T" (T는 테스트 케이스 번호)를 출력한 뒤, 답을 출력한다.

 

내가 쓴 코드

# inorder
# root, N/2번 노드 구하기
def inorder(v):
    global cnt
    # v : 인덱스 번호, N : 노드 수
    if v > N:
        return
    inorder(v * 2)
    lst[v] = cnt
    cnt += 1
    inorder(v * 2 + 1)

T = int(input())
for t in range(T):
    N = int(input())
    lst = [0] * (N + 1)
    cnt = 1     # 값 저장 : 1 ~ N
    inorder(1)
    print(f'#{t+1} {lst[1]} {lst[N//2]}')

어려웡... 나는 인덱스랑 값이랑 반대로 생각하고 풀고있었다.... 먼가 이상했는데, 강사님이 해결해주심!

근데 사실 100% 이해가 되진 않았다,, 한 88% 이해됨. ㅎㅋ 아냐 89% 정도? 헷.

 

출력 결과

#1 4 6
#2 5 2
#3 8 14

728x90
반응형

'Algorithm > Python' 카테고리의 다른 글

[BOJ #1966] 프린터 큐  (0) 2024.02.21
[BOJ #2804] 크로스워드 만들기  (2) 2024.02.20
[BOJ #11721] 열 개씩 끊어 출력하기  (0) 2024.02.20
[SWEA #1231] 중위순회  (1) 2024.02.20
[SWEA #12914] subtree  (1) 2024.02.20