【剑指offer】08二叉树的下一个节点，C++实现

原创博文，转载请注明出处！

# 题目

父节点指向子节点的指针用实线表示，从子节点指向父节点的指针用虚线表示。

# 思路

• 如果节点有右子节点，则右子节点的最左节点是该节点的下一个节点。例如，寻找b的下一个节点的过程（b有右子节点e，e的左子节点是h，且h是e的最左节点，h是b的下一个节点）
• 如果节点无右子节点，但该节点是父节点的左子节点，则父节点是该节点的下一个节点。例如，寻找d的下一个节点的过程（d无右子节点，d是父节点b的左子节点，则b是de的下一个节点）
• 如果节点无右子节点，且该节点是父节点的右子节点，则沿着父节点的指针向上遍历。例如，寻找i的下一个节点的过程（i的父节点e，e是其父节点b的右子节点，节点b是其父节点a的左子节点，节点a是节点i的下一个节点）

# 代码

/*
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 == nullptr)
            return nullptr;

        /* 寻找结果 */
        // 如果节点有右子节点，则右子节点的最左节点是该节点的下一个节点
        // 如果节点无右子节点，但该节点是其父节点的左子节点，则父节点是该节点的下一个节点
        // 如果节点无右子节点，且该节点是其父节点的右子节点，则沿着父节点向上遍历，满足XXX的父节点是其该节点的下一个节点
        TreeLinkNode * res = nullptr;
        if(pNode->right != nullptr)
        {
            TreeLinkNode* node_right = pNode->right;
            while(node_right ->left != nullptr)
            {
                node_right = node_right->left;
            }
            res = node_right;
        }

        else if(pNode->next != nullptr)
        {
            // 当前节点和当前节点的父节点
            TreeLinkNode * current = pNode;
            TreeLinkNode * parent = pNode->next;

            while(parent!=nullptr && current == parent->right)
            {
                current = parent;
                parent = parent ->next;
            }

            res = parent;
        }
        return res;
    }
};

# 复杂度

# 测试用例

• 空节点
• 特殊二叉树（只有一个节点的二叉树、只有左子节点的二叉树、只有右子节点的二叉树、完全二叉树和不完全二叉树）
• 不同位置的节点的下一个节点（下一个节点是当前节点右子节点、右子树的最左子节点、父节点、跨层父节点、无下一个节点）