leetcode【每日一题】1028. 从先序遍历还原二叉树 Java

最新推荐文章于 2022-05-22 14:33:31 发布

宰了那只汤姆猫

最新推荐文章于 2022-05-22 14:33:31 发布

阅读量158

点赞数

分类专栏： leetcode刷题

本文链接：https://blog.csdn.net/qq_43491066/article/details/106825136

版权

leetcode刷题专栏收录该内容

476 篇文章 3 订阅

订阅专栏

题干

我们从二叉树的根节点 root 开始进行深度优先搜索。

在遍历中的每个节点处，我们输出 D 条短划线（其中 D 是该节点的深度），然后输出该节点的值。（如果节点的深度为 D，则其直接子节点的深度为 D + 1。根节点的深度为 0）。

如果节点只有一个子节点，那么保证该子节点为左子节点。

给出遍历输出 S，还原树并返回其根节点 root。

示例 1：

在这里插入图片描述

输入："1-2--3--4-5--6--7"
输出：[1,2,5,3,4,6,7]

示例 2：
在这里插入图片描述

输入："1-2--3---4-5--6---7"
输出：[1,2,5,3,null,6,null,4,null,7]

示例 3：

在这里插入图片描述

输入："1-401--349---90--88"
输出：[1,401,null,349,88,90]

提示：

原始树中的节点数介于 1 和 1000 之间。
每个节点的值介于 1 和 10 ^ 9 之间。

想法

题就是通过字符串生成树
从字符串获得level和数值
得到以后遍历字符串
很容易发现如果A为B的子节点，那么DepthA>DepthB。
所以我们只需要对S字符串进行全局遍历即可。同一个全局变量cur去处理节点值，同时用一个全局变量curD维护当前遍历到的深度，再用一个栈变量depth维护当前节点深度，如果curD>depth则显然全局遍历的下一节点为当前节点的子节点，否则为其长辈节点（叔父节点，祖叔父节点…）。

Java代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
   int cur=0,curD=0;

    public TreeNode recoverFromPreorder(String S) {
        char[] nodes=S.toCharArray();
        int len=nodes.length;
        return dfs(0,nodes,len);

    }

    public TreeNode dfs(int dept,char[] nodes,int len){
        int val=0;
        for(;cur<len&&nodes[cur]!='-';cur++){
            val=val*10+nodes[cur]-'0';
        }
        curD=0;
        for(;cur<nodes.length&&nodes[cur]=='-';cur++,curD++);
        TreeNode r = new TreeNode(val);
        if(curD>dept)r.left = dfs(curD,nodes,len);
        if(curD>dept)r.right = dfs(curD,nodes,len);
        return r;


    }
}