带有父节点指针的二叉树找前驱节点和后继节点

带有父节点指针的二叉树找前驱节点和后继节点

题目：给一个正常的二叉树节点类型加一个指向父节点的指针parent。找其中某一个节点的前驱节点和后继节点

*分析：所为前驱节点和后继节点就是找中序遍历。

答：找出一个结点出现的所有可能性并进行操作。可分为下面几种出现的情况

•  1. 如果该结点无左右子树，判读该结点是否有父节点？
  

•    1.1 如果有父节点，则判断该节点为父节点的左、右子树？
  

•        1.1.1 如果为左子树则该父节点就是其后继节点
  

•        1.1.2 如果为右子树，那么向上找，找出第一个节点有左孩子的那个节点就为其后继节点
  

•    1.2 如果没有父节点，则这个节点即没有父节点也无左右孩子节点，则为根节点，只需返回null或者返回自己。
  

•  2. 如果该节点有左孩子，无右孩子，如果此节点是其父节点的左孩子那么返回父节点，如果此节点是父节点的右孩子则返回该节点左孩子的最右孩子
  

•  3. 如果该节点有右孩子，有无左孩子都可以。即只需返回其右节点的最左节点，就是其后继节点
  

在这里其实有两种方法去写，其实最简单的就是直接递归写中序遍历的代码，然后在根据序列找出前驱和后继，这种方法应该是很多同学第一时间就想到的，也包括我自己。但是如果考虑到算法效率的话，这种写法肯定不是一种时间复杂度低的方法。所有我们注意到题目给我们一个父节点指针，所以肯定有其他方法来写。所以我给大家介绍一下我自己写的第二种方法。

//给一个二叉树节点加一个指向父节点的指针parent

//所谓找后继结点或者是前驱结点就是找中序遍历序列
//找该节点的后继结点
/*
 *  分析：找出一个结点出现的所有可能性并进行操作。可分为下面几种出现的情况
 *      1. 如果该结点无左右子树，判读该结点是否有父节点？
 *        1.1 如果有父节点，则判断该节点为父节点的左、右子树？
 *            1.1.1 如果为左子树则该父节点就是其后继节点
 *            1.1.2 如果为右子树，那么向上找，找出第一个节点有左孩子的那个节点就为其后继节点
 *        1.2 如果没有父节点，则这个节点即没有父节点也无左右孩子节点，则为根节点，只需返回null或者返回自己。
 *      2. 如果该节点有左孩子，无右孩子，如果此节点是其父节点的左孩子那么返回父节点，如果此节点是父节点的右孩子则返回该节点左孩子的最右孩子
 *      3. 如果该节点有右孩子，有无左孩子都可以。即只需返回其右节点的最左节点，就是其后继节点
 *      
 *    总结出这几种可能后，就将返回节点相同的一起写，这样可以优化一下代码。
 * 
 * 
 * */
public class GetLaterNode {
	static class Node {
		public int value;
		Node leftChild;
		Node rightChild;
		Node parent;//

		public Node(int data, Node lc, Node rc,Node p) {
			this.value = data;
			this.leftChild = lc;
			this.rightChild = rc;
			this.parent = p;
		}
	}

	public static void main(String[] args) {
		//初始化一颗树父指针会单独赋值
		Node node4 = new Node(4, null, null,null);
		Node node5 = new Node(5, null, null,null);
		Node node2 = new Node(2, node4, node5,null);
		Node node3 = new Node(3, null, null,null);
		Node node1 = new Node(1, node2, node3,null);
		//给父指针赋值
		node4.parent = node2;
		node5.parent = node2;
		node2.parent = node1;
		node1.parent = null;
		node3.parent = node1;
		
		//如果打印本身 就代表无后继或者前驱
		System.out.println(getLaterNode(node3).value);//打印后继结点
		System.out.println(getFrontNode(node3).value);//打印后继节点
	}
     
	//找后继节点
	public static Node getLaterNode(Node node) {
		if (node == null)
			return node;
		
		//定义parent = node.parent
		Node parent = node.parent;
		
		// 如果有右子树找到右子树最左节点为该子树的后继结点
		if (node.rightChild != null) {
			return getMostLeft(node.rightChild);
		}
		
		// 如果左右子树都无
		else if(node.rightChild==null && node.leftChild==null){
			if(parent!=null&&parent.leftChild==node){
				return parent;
			}
			
		else if(parent!=null&&parent.rightChild==node){
			    //保存变量，如果while没有查到那个合适的元素，不能返回空指针异常。
			    Node node1 = node;
				while (parent != null && parent.rightChild == node) {
					node = parent;
					parent = node.parent;
				}
				return parent==null ? node1 : parent;
			}
		}
		
		else {// 如果无右子树那就看是否有父节点，向上找，找出第一个节点有左孩子那个节点
			
			while (parent != null && parent.rightChild == node) {
				node = parent;
				parent = node.parent;
			}
			return node;
		}
		return node;
	}
    //找最左节点
	public static Node getMostLeft(Node node) {
		if (node == null)
			return node;// 返回自己表示没有后继
		while (node.leftChild != null) {
			node = node.leftChild;
		}
		return node;
	}
	
	// 找前驱结点
	/*
	 * 分析：与后继节点的分析基本类似
	 *     
	 * */
	public static Node getFrontNode(Node node) {
		Node parent = node.parent;
		if (node == null)
			return node;
		else if (node.leftChild != null && node.rightChild == null)
			return getMostRight(node.leftChild);// 当左孩子不为空，右孩子为空时 应该找左孩子的最右结点
		else if (node.leftChild == null && node.rightChild != null)
			return node;// 返回自己 表示没有前驱
		else if (node.leftChild != null && node.rightChild != null)
			return node.leftChild;
		else if (node.leftChild == null && node.rightChild ==null){
			Node node1 =node;//避免后面while循环找不到那个节点，返回空的情况；
			//如果该节点为父亲节点的左孩子，向上找第一个节点有右孩子的那个节点
			while(parent != null && parent.rightChild != node) {
				node = parent;
				parent = node.parent;
			}
			return parent==null ? node1 : parent;
		}
			
		return node;
	}
	// 找左孩子的最右结点
	public static Node getMostRight(Node node) {
		if (node == null)
			return node;

		while (node.rightChild != null) {
			node = node.rightChild;
		}

		return node;
	}
}