문제 설명
https://leetcode.com/problems/populating-next-right-pointers-in-each-node-ii/
자세한 문제 설명은 위 링크 참고. 이진 트리의 각 노드의 next를 모두 오른쪽에 있는 노드를 가리키도록 변경하는 문제다.
BFS를 적용하면 적어도 같은 레벨에 있는 노드에 대해서는 오른쪽 노드를 식별하기 쉽지만, 문제는 레벨이 다른 노드에서다. 단순히 BFS를 사용하면 이전 레벨의 맨 마지막 노드를 다음 레벨의 맨 왼쪽 노드를 가리키게 된다.
이 문제를 해결하기 위해서 아래와 같은 방법을 사용했다.
- BFS로 순회하면서 visited에 노드를 넣을떄 부모 노드 정보를 함께 넣어주었다.
- 순회가 끝난 visited를 탐색하면서 부모노드 정보가 달라지는 지점을 파악하여 맨 오른쪽 노드에 대한 처리를 해준다.
풀이 코드
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
class Solution:
def connect(self, root: 'Node') -> 'Node':
if not root:
return root
visited = []
queue = []
queue.append({
"node" : root,
"level" : 1})
while queue:
elem = queue.pop(0)
visited.append(elem)
if elem["node"].left:
queue.append({
"node": elem["node"].left,
"level": elem["level"] + 1})
if elem["node"].right:
queue.append({
"node": elem["node"].right,
"level": elem["level"] + 1})
# 같은 레벨에 대해서 right 바꿔줌
# 첫 노드는 right를 Null 로하고
elem = visited.pop(0)
elem["node"].next = None
level = 2
for i in range(0, len(visited)-1):
ex = visited[i]
present = visited[i+1]
print("Node : ", ex["node"].val)
print("level : ", ex["level"])
print("Next Node level", present["level"])
# 레벨이 올라가기 전까지는 Node의 right는 그 다음 Node로 함
if ex["level"] != present["level"]:
print("Set next None")
ex["node"].next = None
else:
ex["node"].next = present["node"]
# 마지막 노드에 대한 처리
if visited:
elem = visited.pop(-1)
elem["node"].next = None
return root