【Java】 剑指offer(6) 重建二叉树
本文参考自《剑指offer》一书,代码采用Java语言。
题目
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1, 2, 4, 7, 3, 5, 6, 8}和中序遍历序列{4, 7, 2, 1, 5, 3, 8, 6},则重建出其二叉树并输出它的头结点。
思路
前序遍历第一个值就是根结点的值,根据该值在中序遍历的位置,可以轻松找出该根结点左右子树的前序遍历和中序遍历,之后又可以用同样方法构建左右子树,所以该题可以采用递归的方法完成。
刚开始思考的时候,想的是构建一个遍历函数,输入为前序和中序遍历的数组,输出为根结点。但是这样的话每次都需要构建子树的数组,非常麻烦。
之后想到,该函数的输入不一定要用数组,因为最初的前序和中序遍历数组已经有了,就直接用该数组的下标来表示子树的数组即可。
即构建函数construct(int[] pre, int[] in, int pStart, int pEnd, int iStart, int iEnd),pre和in始终用最初前序遍历和中序遍历的数组代入,pStart、pEnd代表当前树的前序数组开始和结束位置,iStart、iEnd代表中序数组开始和结束位置。
测试用例
1.正常二叉树
2.左斜树
3.右斜树
4.单个结点
5.数组为空
6.前序与中序不匹配
完整Java代码
(含测试代码)
/**
*
* @Description 重建二叉树
*
* @author yongh
* @date 2018年9月12日 下午4:35:19
*/ // 题目:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输
// 入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,
// 2, 4, 7, 3, 5, 6, 8}和中序遍历序列{4, 7, 2, 1, 5, 3, 8, 6},则重建出
// 二叉树并输出它的头结点。
public class ConstructBinaryTree {
class TreeNode {
int val; public TreeNode(int val) {
this.val = val;
} TreeNode left;
TreeNode right;
} public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
if (pre == null || in == null || pre.length <= 0 || in.length <= 0 || pre.length != in.length) {
throw new RuntimeException("数组不符合规范!");
}
return construct(pre, in, 0, pre.length - 1, 0, in.length - 1);
} /**
*
* @Description 由前序遍历序列和中序遍历序列得到根结点
* pre、in:始终用最初的前序遍历和中序遍历数组代入
* pStart、pEnd:当前树的前序数组开始和结束位置
* iStart、iEnd:中序数组开始和结束位置
*/
public TreeNode construct(int[] pre, int[] in, int pStart, int pEnd, int iStart, int iEnd) {
TreeNode root = new TreeNode(pre[pStart]);
if (pStart == pEnd && iStart == iEnd) {
if (pre[pStart] != in[iStart])
throw new RuntimeException("数组不符合规范!");
return root;
}
int index = iStart; // 用于记录中序遍历序列中根结点的位置
while (root.val != in[index] && index <= iEnd) {
index++;
}
if (index > iEnd)
throw new RuntimeException("数组不符合规范!");
int leftLength = index - iStart;
if (leftLength > 0) {
root.left = construct(pre, in, pStart + 1, pStart + leftLength, iStart, index - 1);
}
if (leftLength < iEnd - iStart) {
root.right = construct(pre, in, pStart + leftLength + 1, pEnd, index + 1, iEnd);
}
return root;
} private void preOrderTraverse(TreeNode node) {
if (node == null)
return;
System.out.print(node.val);
preOrderTraverse(node.left);
preOrderTraverse(node.right);
} private void inOrderTraverse(TreeNode node) {
if (node == null)
return;
inOrderTraverse(node.left);
System.out.print(node.val);
inOrderTraverse(node.right);
} /**
* 正常二叉树
*/
public void test1() {
int[] pre = { 1, 2, 4, 7, 3, 5, 6, 8 };
int[] in = { 4, 7, 2, 1, 5, 3, 8, 6 };
TreeNode root = reConstructBinaryTree(pre, in);
System.out.print("test1:");
preOrderTraverse(root);
System.out.print("//");
inOrderTraverse(root);
System.out.println();
} /**
* 左斜树
*/
public void test2() {
int[] pre = { 1, 2, 3, 4, 5 };
int[] in = { 5, 4, 3, 2, 1 };
TreeNode root = reConstructBinaryTree(pre, in);
System.out.print("test2:");
preOrderTraverse(root);
System.out.print("//");
inOrderTraverse(root);
System.out.println();
} /**
* 右斜树
*/
public void test3() {
int[] pre = { 1, 2, 3, 4, 5 };
int[] in = { 1, 2, 3, 4, 5 };
TreeNode root = reConstructBinaryTree(pre, in);
System.out.print("test3:");
preOrderTraverse(root);
System.out.print("//");
inOrderTraverse(root);
System.out.println();
} /**
* 单个结点
*/
public void test4() {
int[] pre = { 1 };
int[] in = { 1 };
TreeNode root = reConstructBinaryTree(pre, in);
System.out.print("test4:");
preOrderTraverse(root);
System.out.print("//");
inOrderTraverse(root);
System.out.println();
} /**
* 数组为空
*/
public void test5() {
int[] pre = {};
int[] in = {};
TreeNode root = reConstructBinaryTree(pre, in);
System.out.print("test5:");
preOrderTraverse(root);
System.out.print("//");
inOrderTraverse(root);
System.out.println();
} public static void main(String[] args) {
ConstructBinaryTree demo = new ConstructBinaryTree();
demo.test1();
demo.test2();
demo.test3();
demo.test4();
demo.test5();
}
}
test1://
test2://
test3://
test4://
Exception in thread "main" java.lang.RuntimeException: 数组不符合规范!
ConstructBinaryTree
收获
1.在递归问题中,代码可以用下标表示的就用下标表示,不用重新构建新的数组。
2.数组为空与数组为null不是一回事。
【Java】 剑指offer(6) 重建二叉树的更多相关文章
- 剑指Offer:重建二叉树【7】
剑指Offer:重建二叉树[7] 题目描述 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树.假设输入的前序遍历和中序遍历的结果中都不含重复的数字.例如输入前序遍历序列{1,2,4,7,3,5 ...
- 《剑指offer》重建二叉树
本题来自<剑指offer> 重构二叉树 题目: 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树.假设输入的前序遍历和中序遍历的结果中都不含重复的数字.例如输入前序遍历序列{1,2 ...
- Go语言实现:【剑指offer】重建二叉树
该题目来源于牛客网<剑指offer>专题. 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树.假设输入的前序遍历和中序遍历的结果中都不含重复的数字.例如输入前序遍历序列{1,2,4 ...
- 剑指offer之重建二叉树
1.问题描述:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树.假设输入的前序遍历和中序遍历的结果中都不含重复的数字. 例如输入前序遍历序列pre {1,2,4,7,3,5,6, ...
- 剑指OFFER之重建二叉树(九度OJ1385)
题目描述: 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树.假设输入的前序遍历和中序遍历的结果中都不含重复的数字.例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7 ...
- 剑指offer:重建二叉树
重建二叉树的前置知识: 0.遍历二叉树: (1)前序遍历:根左右 --> 先访问根节点,再前序遍历左子树,最后前序遍历右子树: (2)中序遍历:左根右 --> 先中序遍历左子树,再访问根节 ...
- 剑指Offer 4. 重建二叉树 (二叉树)
题目描述 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树.假设输入的前序遍历和中序遍历的结果中都不含重复的数字.例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7, ...
- 【剑指offer】重建二叉树
一.题目: 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树.假设输入的前序遍历和中序遍历的结果中都不含重复的数字.例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7 ...
- 剑指offer——04重建二叉树(Python3)
思路:在数据结构中,有一个条件反射,谈及二叉树,就递归.所以在实现重建二叉树时,也应该用到递归的思想. 在前序遍历中,根节点处于第一个:在中序遍历中,根节点的左边为左子树节点,根节点右边为右子树节点. ...
随机推荐
- PHP 获取给定时间的周日时间或月末时间或每天
//获取给定时间的周日时间或月末时间或每天 private function endDate ($time, $type = 1, $openTime = '20:00:00') { if ($typ ...
- mysql 原理 ~ 事务隔离机制
简介: 事务隔离知多少内容 一 基础知识 1 事务特性 ACID A 原子性 C 一致性 I 隔离性 D 持久性 2 并行事务出现的问题 1 脏读 读取了其他事务未提交的数据 ...
- windws 下 sublime Text 3 ·安装的安装与激活
下载sublime 我们可以到官网进行下载对应的版本 https://www.sublimetext.com/3 如下是官网的内容(我选择的是Windows 64 bit). Sublime Text ...
- Maven整合Spring3.0+Mybatis3.2+Struts2.3+查找坐标+jar包依赖(五)
依赖传递 只添加了一个struts2-core依赖,发现项目中出现了很多jar,这种情况 叫 依赖传递
- Vim中自动在程序起始处添加版权和作者信息
在编写程序的时候,经常需要在程序开始写上程序的简要介绍和作者信息,如下: 这种信息,除了文件名和修改时间可能经常发生变化外,其他基本不变,可以在程序开始自动加入,方法就是在家目录下的.vimrc中写入 ...
- js代码解析原则
js引擎在读取js代码时会进行两个步骤,第一个步骤是解释,第二个步骤是执行. 解释就是先通篇扫描所有的Js代码,然后把所有声明提升到顶端,第二步是执行,执行就是执行代码的操作. 例: 例子1: < ...
- Reverse Words in a String I & Reverse Words in a String II
Reverse Words in a String I Given an input string, reverse the string word by word. For example,Give ...
- 无责任共享 Coursera、Udacity 等课程视频(转载)
转载链接:https://www.zybuluo.com/illuz/note/71868 B站计划:https://www.zybuluo.com/illuz/note/832995#cs基础课程
- Linux内核驱动--硬件访问I/O【原创】
寄存器与内存 寄存器与内存的区别在哪里呢? 寄存器和RAM的主要不同在于寄存器操作有副作用(side effect或边际效果): 读取某个地址时可能导致该地址内容发生变化,比如很多设备的中断状态寄存器 ...
- ES系列三、基本知识准备
一.基础概念 1.索引 索引(index)是elasticsearch的一个逻辑存储,可以理解为关系型数据库中的数据库,es可以把索引数据存放到一台服务器上,也可以sharding后存到多台服务器上, ...