「每日LeetCode」2021年1月17日

本文最后更新于:2023年3月19日 晚上

Lt897. 递增顺序查找树

897. 递增顺序查找树

给你一个树,请你 按中序遍历 重新排列树,使树中最左边的结点现在是树的根,并且每个结点没有左子结点,只有一个右子结点。
示例 :

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
输入:[5,3,6,2,4,null,8,1,null,null,null,7,9]
5
/ \
3 6
/ \ \
2 4 8
/ / \
1 7 9
输出:[1,null,2,null,3,null,4,null,5,null,6,null,7,null,8,null,9]
1
\
2
\
3
\
4
\
5
\
6
\
7
\
8
\
9

提示:

  1. 给定树中的结点数介于 1100 之间。
  2. 每个结点都有一个从 01000 范围内的唯一整数值。

思路

中序遍历得到递增数组,再根据数组生成递增树

解答

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
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
* @return {TreeNode}
*/
var increasingBST = function (root) {
const arr = [];
const visit = (root) => {
if (!root) return;
visit(root.left);
arr.push(root.val);
visit(root.right);
};
visit(root);
const res = new TreeNode(arr.shift());
let temp = res;
while (arr.length) {
temp.right = new TreeNode(arr.shift());
temp = temp.right;
}
return res;
};