题目:
给你一棵二叉搜索树,请 按中序遍历 将其重新排列为一棵递增顺序搜索树,
使树中最左边的节点成为树的根节点,并且每个节点没有左子节点,只有一个右子节点。
输入: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]
输入:root = [5,1,7]
输出:[1,null,5,null,7]
提示:
树中节点数的取值范围是 [1, 100]
0 <= Node.val <= 1000
--------------------
思路:
展平二叉搜索树,也就是对二叉搜索树进行中序遍历,然后将各个节点的指针进行修改,
从而形成一棵只有右子树的树,解题思路如下:
设置一个辅助头节点 head : 指向重排后的树。
对树进行中序遍历,将遍历到的节点插入重排后的树head的最后一个节点的右子树,
同时将当前遍历到的节点的左子树设置为 null。
遍历完后,直接返回 head.right 就是重排后的树的根节点。
注意:由于插入操作需要知道重排后的树 head 的最后一个节点,
所以可以设置一个节点 cur 指向重排后的树的最后一个节点。
-----------------------
/**
* 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 {
private TreeNode cur;//cur 指向重排后的树的最后一个节点
public TreeNode increasingBST(TreeNode root) {
TreeNode head = new TreeNode(0, null, null);//创建一个辅助头节点,cur指向辅助头节点
cur = head;//初始化
dfs(root);//DFS 中序遍历
return head.right;//返回树的第一个节点
}
private void dfs(TreeNode node) {
if (node == null) return;//递归终止条件: node为空
dfs(node.left);//中序遍历: 左--中--右
cur.right = node;//将当前节点插入到重排后树的右子树,同时更新cur这个重排树的最后一个节点
cur = cur.right;
node.left = null;//将节点的左边节点设置为null
dfs(node.right);
}
}
LC