重建二叉树题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
根据数据结构课程Tree Traversals Again该题的思路,即已知前序和中序,写出后序。所以只给出前序遍历第一个元素,中序遍历第一个元素和遍历的个数。
比较常见的解法是给出前序遍历序列数组的第一个元素和最后一个元素;
中序遍历序列数组的第一个元素和最后一个元素。
这里将两种思路的代码一并给出
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
int n = pre.length;
return solve(pre,in,0,0,n);
}
public TreeNode solve(int [] pre,int [] in,int preL,int inL,int n) {
if(n==0) return null;
TreeNode tree = new TreeNode(pre[preL]);
if(n == 1) {
return tree;
}
int i;
for(i = 0;i<n;i++) {
if( in[inL+i] == tree.val) break;
}
int L = i; int R = n-i-1;
if(L>0) {
tree.left = solve(pre,in,preL+1,inL,L);
}
if(i<n) {
tree.right = solve(pre,in,preL+L+1,inL+L+1,R);
}
return tree;
}
}
需要注意的是,这里必须给出
更多人使用的思路:
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
public class TestRecoverBinaryTree {
public TreeNode reConstructBinaryTree(int [] preOrder,int [] inOrder)
{
int pLen = preOrder.length;
int iLen = inOrder.length;
if(pLen==0 && iLen==0)
{
return null;
}
return btConstruct( preOrder, inOrder, 0, pLen-1,0, iLen-1);
}
//构建方法,pStart和pEnd分别是前序遍历序列数组的第一个元素和最后一个元素;
//iStart和iEnd分别是中序遍历序列数组的第一个元素和最后一个元素。
public TreeNode btConstruct(int[] preOrder, int[] inOrder, int pStart, int pEnd,int iStart,int iEnd)
{
//建立根节点
TreeNode tree = new TreeNode(preOrder[pStart]);
tree.left = null;
tree.right = null;
if(pStart == pEnd && iStart == iEnd)
{
return tree;
}
int root = 0;
//找中序遍历中的根节点
for(root=iStart; root<iEnd; root++)
{
if(preOrder[pStart] == inOrder[root])
{
break;
}
}
//划分左右子树
int leftLength = root - iStart;//左子树
int rightLength = iEnd - root;//右子树
//遍历左子树
if(leftLength>0)
{
tree.left = btConstruct(preOrder, inOrder, pStart+1, pStart+leftLength, iStart, root-1);
}
//遍历右子树
if(rightLength>0)
{
tree.right = btConstruct(preOrder, inOrder, pStart+leftLength+1, pEnd, root+1, iEnd);
}
return tree;
}
}
————————————————
版权声明:本文为CSDN博主「roycon」的原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/roycon/article/details/79772639
二叉搜索树的第K个结点题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。
一开始没有仔细审题我以为是返回值就行,所以想的是将树的值写入一个数组中,
然后排序,然后找到第k个值,结果要求返回一个结点,所以整个傻眼掉,
陷入僵局,搜索之后发现,原来解题关键在于二叉搜索树的定义,“每个结点的值,
比左子树大,比右子树小”所以直接中序排列就可以得到从小到大的顺序。
然后结合树的遍历方法中(三种遍历方法介绍https://blog.csdn.net/coder__666/article/details/80349039),非递归的方法,加入一个计数
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
import java.util.Stack;
public class Solution {
TreeNode KthNode(TreeNode pRoot, int k)
{
Stack<TreeNode> treestack = new Stack<TreeNode>();
int count = 0;
while(pRoot!=null || !treestack.isEmpty()) {
while(pRoot != null) {
treestack.push(pRoot);
pRoot = pRoot.left;
}
count++;
if( count == k) {
return treestack.pop();
}
else
{
pRoot = treestack.pop();
pRoot = pRoot.right;
}
}
return null;
}
}