加油
中序遍历之后生成新的树
算法
题目要求我们返回按照中序遍历的结果改造而成的、只有右节点的等价二叉搜索树。我们可以进行如下操作:
先对输入的二叉搜索树执行中序遍历,将结果保存到一个列表中;
然后根据列表中的节点值,创建等价的只含有右节点的二叉搜索树,其过程等价于根据节点值创建一个链表。
class Solution {
public TreeNode increasingBST(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
inorder(root, res);
TreeNode dummyNode = new TreeNode(-1);
TreeNode currNode = dummyNode;
for (int value : res) {
currNode.right = new TreeNode(value);
currNode = currNode.right;
}
return dummyNode.right;
}
public void inorder(TreeNode node, List<Integer> res) {
if (node == null) {
return;
}
inorder(node.left, res);
res.add(node.val);
inorder(node.right, res);
}
}
第二个
展平二叉搜索树,也就是对二叉搜索树进行中序遍历,然后将各个节点的指针进行修改,从而形成一棵只有右子树的树,解题思路如下:
设置一个辅助头节点 head : 指向重排后的树。
对树进行中序遍历,将遍历到的节点插入重排后的树head的最后一个节点的右子树,同时将当前遍历到的节点的左子树设置为 null。
遍历完后,直接返回 head.right 就是重排后的树的根节点。
class Solution {
// cur 指向重排后的树的最后一个节点
private TreeNode cur;
public TreeNode increasingBST(TreeNode root) {
// 创建一个辅助头节点,cur 指向辅助头节点
TreeNode head = new TreeNode(0,null, null);
cur = head;
// 进行中序遍历 DFS
dfs(root);
// 返回树的第一个节点。
return head.right;
}
private void dfs(TreeNode node){
// 递归终止条件:node 为空
if(node == null) return;
// 中序遍历:左->中->右
dfs(node.left);
// 将当前节点插入重排的树的右子树中,同时更新重排树的最后一个节点。
cur.right = node;
cur = cur.right;
// 将节点的左边节点设置为 null。
node.left = null;
dfs(node.right);
}
}