897.递归顺序查找树

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

示例 :

输入:[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 范围内的唯一整数值。

这道题目思路很简单,用中序遍历即可,在中序遍历的过程中把当前遍历的节点变成上一个节点的右节点即可,但是做这个题目的时候卡住了一会,原因是返回的时候直接返回了cur,这就会导致返回的是最后一个节点的右节点,为null

    class Solution {
        public TreeNode increasingBST(TreeNode root) {
            if(root == null) {
                return null;
            }
            LinkedList<TreeNode> list = new LinkedList<>();
            TreeNode newRoot = new TreeNode(0);
            TreeNode cur = newRoot;
            //怎么返回头节点是一个值得注意的点
            while(root != null || !list.isEmpty()) {
                if(root != null) {
                    list.add(root);
                    root = root.left;
                }else {
                    TreeNode node = list.pollLast();
                    cur.right = new TreeNode(node.val);
                    root = node.right;
                    cur = cur.right;
                }
            }
            return newRoot.right;
        }
    }

解决办法也很简单,用两个节点,一个用来返回,一个用来遍历。

猜你喜欢

转载自www.cnblogs.com/Jiewl/p/12544278.html