Skip to content

Commit eb67723

Browse files
Create taurus09318976.py
1 parent 51ec641 commit eb67723

File tree

1 file changed

+89
-0
lines changed

1 file changed

+89
-0
lines changed
Lines changed: 89 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,89 @@
1+
"""
2+
문제의도: 이 문제는 트리 순회의 특성을 이해하고 재귀적 분할정복을 연습하는 문제임.
3+
-> preorder로 루트 찾고, inorder로 좌우 나누고, 재귀로 반복
4+
- 전위 순회(preorder): 루트 -> 왼쪽 -> 오른쪽
5+
- 중위 순회(inorder): 왼쪽 -> 루트 -> 오른쪽
6+
7+
해결 방법:
8+
1. preorder의 첫 번째 원소가 현재 트리의 루트임을 이용
9+
2. inorder에서 루트의 위치를 찾아 왼쪽/오른쪽 서브트리 분할
10+
3. preorder에서도 대응하는 부분을 분할
11+
4. 왼쪽/오른쪽 서브트리에 대해 재귀적으로 같은 과정 반복
12+
-> - 루트를 알면 inorder에서 왼쪽/오른쪽을 구분할 수 있음!
13+
14+
시간복잡도: O(n²)
15+
- index() 메서드: 배열에서 값을 찾기 위해 최악의 경우 전체 배열을 순회 → O(n)
16+
- 재귀 호출 횟수: 모든 노드에 대해 한 번씩 호출 → n번
17+
- 전체 시간복잡도: n번의 호출 × 각 호출마다 O(n) = O(n²)
18+
19+
공간복잡도: O(n)
20+
- 재귀 호출 스택의 깊이: O(n) (편향 트리일 때)
21+
- 새로 생성하는 배열들: O(n)
22+
23+
예시 설명 :
24+
예시1의 경우
25+
26+
1단계: preorder의 첫 번째는 항상 루트
27+
[3, 9, 20, 15, 7] → 3이 루트
28+
29+
2단계: inorder에서 루트 위치로 좌우 분할
30+
[9, 3, 15, 20, 7] → 3 기준으로 [9] | [15, 20, 7]
31+
- 왼쪽: [9] (인덱스 0까지)
32+
- 오른쪽: [15, 20, 7] (인덱스 2부터)
33+
34+
3단계: 왼쪽 서브트리 크기로 preorder 분할
35+
왼쪽 크기가 1이면 preorder에서도 1개만 가져옴
36+
- 루트 제외: [9, 20, 15, 7]
37+
- 왼쪽 크기(1)만큼: [9]
38+
- 나머지: [20, 15, 7]
39+
40+
4단계: 재귀적으로 같은 과정 반복
41+
- 왼쪽: buildTree([9], [9]) → 노드 9
42+
- 오른쪽: buildTree([20, 15, 7], [15, 20, 7]) → 서브트리
43+
44+
최종 결과:
45+
3
46+
/ \
47+
9 20
48+
/ \
49+
15 7
50+
51+
52+
주의사항:
53+
- 배열 슬라이싱 범위 조심하기
54+
- 빈 배열일 때 None 반환하기
55+
- inorder에서 루트 위치 정확히 찾기
56+
57+
"""
58+
59+
60+
class Solution:
61+
def buildTree(self, preorder, inorder):
62+
# 1. 기본 케이스: 빈 배열이면 None 반환
63+
if not preorder or not inorder:
64+
return None
65+
66+
# 2. preorder의 첫 번째가 루트
67+
root_val = preorder[0]
68+
root = TreeNode(root_val)
69+
70+
# 3. inorder에서 루트 위치 찾기
71+
root_index = inorder.index(root_val)
72+
73+
# 4. inorder를 루트 기준으로 분할
74+
left_inorder = inorder[:root_index]
75+
right_inorder = inorder[root_index + 1:]
76+
77+
# 5. preorder도 해당 크기만큼 분할
78+
left_preorder = preorder[1:1 + len(left_inorder)]
79+
right_preorder = preorder[1 + len(left_inorder):]
80+
81+
# 6. 재귀적으로 왼쪽과 오른쪽 서브트리 구성
82+
root.left = self.buildTree(left_preorder, left_inorder)
83+
root.right = self.buildTree(right_preorder, right_inorder)
84+
85+
return root
86+
87+
88+
89+

0 commit comments

Comments
 (0)