题目:
给你一棵二叉搜索树,请 按中序遍历 将其重新排列为一棵递增顺序搜索树,
使树中最左边的节点成为树的根节点,并且每个节点没有左子节点,只有一个右子节点。
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/0c586148067ec6cc435cd886cce47bac.png)
输入:root = [5,3,6,2,4,null,8,1,null,null,null,7,9]
输出:[1,null,2,null,3,null,4,null,5,null,6,null,7,null,8,null,9]
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/b446e355957e575e2c8892d22e630d9d.png)
输入:root = [5,1,7]
输出:[1,null,5,null,7]
提示:
树中节点数的取值范围是 [1, 100]
0 <= Node.val <= 1000
--------------------
思路:
展平二叉搜索树,也就是对二叉搜索树进行中序遍历,然后将各个节点的指针进行修改,
从而形成一棵只有右子树的树,解题思路如下:
设置一个辅助头节点 head : 指向重排后的树。
对树进行中序遍历,将遍历到的节点插入重排后的树head的最后一个节点的右子树,
同时将当前遍历到的节点的左子树设置为 null。
遍历完后,直接返回 head.right 就是重排后的树的根节点。
注意:由于插入操作需要知道重排后的树 head 的最后一个节点,
所以可以设置一个节点 cur 指向重排后的树的最后一个节点。
-----------------------
class Solution {
private TreeNode cur;
public TreeNode increasingBST(TreeNode root) {
TreeNode head = new TreeNode(0, null, null);
cur = head;
dfs(root);
return head.right;
}
private void dfs(TreeNode node) {
if (node == null) return;
dfs(node.left);
cur.right = node;
cur = cur.right;
node.left = null;
dfs(node.right);
}
}
LC