题目:给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
分析:
根据中序遍历的特点,要找到一个节点的下一个节点无非就是三种情况:
1、有右子树,这时只需要把其右孩子作为下一个遍历的(并不是要找的)节点,然后沿着该节点的左子树(如果有的话)出发,直到遇到叶子节点,那么该叶子节点就是其下一个要找的节点;
2、没有右子树,则判断该节点是否是其父节点的左孩子,如果是则其下一个要找的节点是其父节点;
3、如果不是其父节点的左孩子,则把其父节点作为下一个遍历的节点,向上回溯,直到找到节点没有父节点或者节点是其父节点的左孩子为止。
综合这三种情况就可以找到二叉树中任意一个节点的下一个节点。
代码如下:
public class FindNextNode { public BinaryTreeNode getNextNode(BinaryTreeNode pNode){ if(pNode==null){ return null; } //如果该节点有右节点 if(pNode.getRightNode()!=null){ BinaryTreeNode tempNode=pNode.getRightNode(); while(tempNode.getLeftNode()!=null){ tempNode=tempNode.getLeftNode(); } return tempNode; } //如果没有右节点,是其父节点的左子节点 if(pNode.getFatherNode()==null){ return null; } if(pNode.getFatherNode().getLeftNode()==pNode){ return pNode.getFatherNode(); } //如果没有右节点,是其父节点的右子节点,一直向上找父节点直到没有父节点 if(pNode.getFatherNode()==null){ return null; } if(pNode.getFatherNode().getRightNode()==pNode){ BinaryTreeNode tempNode=pNode.getFatherNode(); while(tempNode.getFatherNode()==null){ tempNode=tempNode.getFatherNode(); } return tempNode; } return null; } }