剑指offer-题58：二叉树下的一个结点

题目描述

给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。

实验平台：牛客网

---

解决思路：

java:

/*
public class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;

    TreeLinkNode(int val) {
        this.val = val;
    }
}
*/
public class Solution {
    public TreeLinkNode GetNext(TreeLinkNode pNode){
        if (pNode == null) {
            return null;
        } else {
            TreeLinkNode nextNode = null;
            if (pNode.right != null) {
                nextNode = pNode.right;
                while (nextNode.left != null) {
                    nextNode = nextNode.left;
                }
            } else {
                TreeLinkNode currentNode = pNode;
                TreeLinkNode parentNode = pNode.next;
                while (parentNode != null && parentNode.right == currentNode) {
                    currentNode = parentNode;
                    parentNode = currentNode.next;
                }
                nextNode = parentNode;
            }
            return nextNode;
        }
    }
}

python:

略