Cracking the coding interview--Q4.5

最新推荐文章于 2021-11-30 06:58:40 发布

原创最新推荐文章于 2021-11-30 06:58:40 发布 · 1.2k 阅读

0 ·

CC 4.0 BY-SA版权

Cracking the coding interview 专栏收录该内容

100 篇文章

订阅专栏

本文介绍了一种算法，用于在一个具有父节点链接的二叉查找树中找到给定节点的中序遍历后继节点。如果目标节点有右子树，则后继节点为右子树中最左侧的节点；如果没有右子树，则需沿父节点路径上溯，直至找到第一个大于当前节点的祖先。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

原文：

Write an algorithm to find the ‘next’ node (e.g., in-order successor) of a given node in a binary search tree where each node has a link to its parent.

译文：

给定一个二叉查找树的结点，写一个算法去找“next”结点（即中序遍历后的后继结点），其中每一个结点都有指向其父亲的链接。

解答

题目要求的就是寻找中序遍历的后继结点。若该结点有右子树，则后继结点是其右子树的最左孩子；若该结点没有右子树，则需要不断地向上查找该结点的祖先，直到找到第一个比它大的结点为止。

代码如下：

class Q4_5{
	public static TreeNode successor(TreeNode tnode){
		if(tnode==null) return null;
		if(tnode.rchild!=null){
			TreeNode tn=tnode.rchild;
			while(tn.lchild!=null){
				tn=tn.lchild;
			}
			return tn;
		}else{
			TreeNode p=tnode.parent;
			while(p!=null&&p.rchild==tnode){
				tnode=p;
				p=p.parent;
			}
			return p;
		}
	}
	public static void main(String[] args){
		int[]  arr={2,1,9,8,7,10};
		TreeNode root = TreeNode.createBinaryTree(arr);
		TreeNode t = root.rchild.lchild;
		TreeNode n= successor(t);
		System.out.println(n.value);
	}
}

class TreeNode{
	int value;
	TreeNode lchild;
	TreeNode rchild;
	TreeNode parent;
	
	public static void insert(TreeNode tnode,int x,TreeNode p){
		if(tnode==null){
			tnode=new TreeNode();
			tnode.value=x;
			tnode.lchild=null;
			tnode.rchild=null;
			if(p.value>x)
				p.lchild=tnode;
			else
				p.rchild=tnode;
			tnode.parent=p;
			return;
		}
		
		if(x<tnode.value){
			insert(tnode.lchild,x,tnode);
		}else{
			insert(tnode.rchild,x,tnode);
		}
	}
	
	public static TreeNode createBinaryTree(int[] values){
		TreeNode root=new TreeNode();
		root.value=values[0];
		for(int i=1;i<values.length;i++){
			insert(root,values[i],root);
		}
		return root;
	}
}

---EOF---