👀 문제
https://leetcode.com/problems/flatten-binary-tree-to-linked-list/submissions/
👊 도전
1. 설계
- postorder을 이용하여 트리를 재구성한다.
2. 구현
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
/**
*
* @author HEESOO
*
*/
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
TreeNode prev=null;
public void flatten(TreeNode root) {
if(root==null) return;
flatten(root.right);
flatten(root.left);
root.right=prev;
root.left=null;
prev=root;
}
}
3. 결과
🤟 성공 🤟
4. 설명
- postorder을 이용한다
- 후위순회는 RLV로 방문한다.
- 문제의 예제에서 후위순회로 방문하면 [6,5,4,3,2,1]이므로 트리의 밑에서 위로 올라가며 오른쪽으로 flatten된 트리를 만들면 된다.
- 재귀를 통해 오른쪽 끝까지 이동한 후, 마지막 노드 root의 right는 이전 prev, left는 null로 바꾼다.
- 이제 root의 위로 올라가므로 prev를 root로 갱신한다.