剑指offer之二叉树的下一个结点

题目描述

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

解题思路

中序遍历的特点，左根右，分三种情况。

1. 如果一个树有右子树，则返回右子树的最左端的左子树结点。
2. 如果当前结点是其父节点的左子树，返回父节点
3. 如果当前结点是其父节点的右子树，返回第一个其父节点的左子树是那个结点的父结点。

Code

/*
struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *next;
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {
        
    }
};
*/
class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if(!pNode) return pNode;
        TreeLinkNode *pNext = nullptr;
        if(pNode->right) {
            pNext = pNode->right;
            while(pNext->left) {
                pNext = pNext->left;
            }
        } else if(pNode->next && pNode->next->left == pNode) {
            pNext = pNode->next;
        } else if(pNode->next && pNode->next->right == pNode) {
            pNext = pNode;
            while(pNext->next && pNext->next->right == pNext) {
                pNext = pNext->next;
            }
            if(pNext) pNext = pNext->next;
        }
        return pNext;
    }
};

• 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)
    {
        TreeLinkNode result = null;
        if(pNode != null) {
            if(pNode.right != null) {
                result = pNode.right;
                while(result.left != null) {
                    result = result.left;
                }
            } else if(pNode.next != null) {
                result = pNode;
                while(result.next != null && result.next.right == result) {
                    result = result.next;
                }
                result = result.next;
            }
        }
        return result;
    }
}