重建二叉树
题目链接:
根据 前序遍历 和 中序遍历重建二叉树,返回根节点
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | import java.util.*; /** * Definition for binary tree * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ public class Solution { Map<Integer,Integer> map = new HashMap<>(); // 存储中序遍历,方便确定前序遍历得到的根节点在中序遍历中 的位置 public TreeNode reConstructBinaryTree( int [] pre, int [] vin) { for ( int i= 0 ;i<pre.length;i++){ map.put(vin[i],i); } return reConstructTree(pre, 0 , 0 ,pre.length- 1 ); } // 递归找到根节点 // preL : 前序第一个节点 preR:前序最后一个节点 vinL中序遍历第一个节点 public TreeNode reConstructTree( int [] pre, int preL, int vinL, int preR){ // 出口 if (preL>preR){ return null ; } // 返回 TreeNode root = new TreeNode(pre[preL]); // 计算左子树长度 // 根节点在右子树的位置 int vIndex = map.get(root.val); int leftSize = vIndex - vinL; // 找左子树的根 TreeNode left = reConstructTree(pre,preL+ 1 ,vinL,preL+leftSize); // 找右子树的根 TreeNode right = reConstructTree(pre,preL+leftSize+ 1 ,vIndex+ 1 ,preR); root.left = left; root.right = right; return root; } } |
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 全程不用写代码,我用AI程序员写了一个飞机大战
· DeepSeek 开源周回顾「GitHub 热点速览」
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 记一次.NET内存居高不下排查解决与启示
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了