题目描述
给你一棵二叉搜索树的 root ,请你 按中序遍历 将其重新排列为一棵递增顺序搜索树,使树中最左边的节点成为树的根节点,并且每个节点没有左子节点,只有一个右子节点。
样例描述
思路
中序遍历 递归/ 迭代
在中序遍历中维护尾结点即可,单独记录一个尾结点,由于头结点可能空也可能不空,创建一个虚拟头结点

代码
/**
* 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 tail;
public TreeNode increasingBST(TreeNode root) {
//虚拟结点
TreeNode dummy = new TreeNode(-1);
tail = dummy;
dfs(root);
return dummy.right;
}
public void dfs(TreeNode root) {
if (root == null) {
return;
}
dfs(root.left);
//当前结点左结点设置为空
root.left = null;
//然后插入到构造的尾结点后面
tail.right = root;
tail = root;
dfs(root.right);
}
}
迭代版
/**
* 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 {
public TreeNode increasingBST(TreeNode root) {
TreeNode dummy = new TreeNode(-1);
TreeNode tail = dummy;
Deque<TreeNode> stack = new LinkedList<>();
while (!stack.isEmpty() || root != null) {
while (root != null) {
stack.push(root);
root = root.left;
}
root = stack.pop();
root.left = null;
tail.right = root;
tail = root;
root = root.right;
}
return dummy.right;
}
}
本文介绍了如何将一个二叉搜索树按中序遍历的顺序重新排列,使其变为递增顺序搜索树。通过迭代的方法,利用中序遍历的特性,维护一个尾节点,确保每个节点只有一个右子节点。
2580

被折叠的 条评论
为什么被折叠?



