剑指 Offer -- 二叉树的下一个节点(五十七)

二叉树的下一个节点(五十七)

题目描述:

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

代码(已在牛客上 AC)

提供两种写法, 都 AC 了. 由于看完了红黑树, 现在看这道题不是很复杂了. 这个节点设置和红黑树
里面的设置一样, 都有指向父节点的指针. 那么来看看需要注意什么问题.

              1
          /       \
         2         3
       /   \     /   \
      4     5   6     7

中序遍历的结果为: 4 2 5 1 6 3 7

• 对于含有右子树的节点, 比如节点 1, 要先访问右孩子, 然后沿着右孩子的 left 一直移动, 得到节点 1 的下一个节点为 6.
• 而对于没有右子树的节点, 比如节点 5, 它下一个节点为 1, 而为了找到这一个节点, 我们设置 pNode 指向节点 5, 然后朝着根节点移动 pNode(即路径 5 - 2 - 1), 直到 pNode 是某节点(设为 p)的左孩子, 那么 p 就是我们要寻找的下一个节点.

class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode *pNode) {
        // 同时没有右孩子和父节点, 直接返回空.
        if (!pNode || (!pNode->right && !pNode->next)) return nullptr;
        if (pNode->right) {
            auto p = pNode->right;
            while (p->left) p = p->left;
            return p;
        }
        // 如果 pNode 一直是 p 的右孩子, 那么就得沿着路径一直向上查找.
        auto p = pNode->next;
        while (p && pNode == p->right) {
            pNode = p;
            p = p->next;
        }
        // 不需要对 p 进行判断, p 可能为空.
        return p;
    }
};

写法二

思路与上面一样的.

class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode) {
        if (!pNode) return pNode;
        TreeLinkNode *pNext;
        if (pNode->right) { // 如果含有右孩子的话
            pNext = pNode->right;
            while (pNext->left) pNext = pNext->left;
        }
        else {
            //如果没有右孩子, 那么就要一直查找父节点, 之后某节点为父节点的左孩子
            auto pCur = pNode, pPar = pCur->next;
            while (pPar && pPar->right == pCur) {
                pCur = pPar;
                pPar = pCur->next;
            }
            pNext = pPar;
        }
        return pNext;
    }
};