题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
分析
对于二叉树中序遍历来说,某node的下一个节点可以分为以下几种情况:
- node.right 不为 null时,根据中序遍历的定义,下一个节点则是node右子树里最左边的节点。
- node.right 为 null时,考察node是否为node.parent的左节点,如果是的话,node的下一个节点就是node.parent;否则,考察node.parent是否为node.parent.parent的左节点,依次这样向上探索下去。
代码实现
/*function TreeLinkNode(x){ this.val = x; this.left = null; this.right = null; this.next = null; }*/ function GetNext(node) { if(node === null) return null; if(node.right !== null){ node = node.right; while(node.left !== null){ node = node.left; } return node; }else{ while(node.next !== null){ if(node === node.next.left) return node.next; node = node.next; } } return null; }