【剑指offer】面试题8:二叉树的下一个节点【C++版本】

总结的部分题目思路与代码，待完善。
【剑指offer-第二版】部分题目与解答【C++版本】

题目：

二叉树的下一个节点

给定一颗二叉树和其中的一个节点，如何找出中序遍历序列的下一个节点？树中的节点除了指向左右子树的指针，还有一个指向父节点的指针。树节点定义如下：

struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *next;
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {}
};

解题思路：

1.如果该节点有右子树，则它的下一个节点就是其右子树的最左子节点。
2.如果该节点没有右子树，但是它本身是其父节点的左子节点，那么它的下一个节点就是其父节点。
3.如果该节点没有右子树，并且是其父节点的右子节点，那么它的下一个节点就需要向上不断找父节点，直到找到一个节点是其父节点的左子节点，那么该节点的父节点就要我们要找的下一个节点。
4.如果上面的方法都没有找到下一个节点，说明该节点不存在下一个节点，返回nullptr。

可以AC的解法【C++版本】

#include <vector>
#include <iostream>

using namespace std;

struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *next;
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {}
};

TreeLinkNode* GetNext(TreeLinkNode* pNode);

int main()
{

    system("pause");
    return 0;
}

TreeLinkNode* GetNext(TreeLinkNode* pNode) {
    if (pNode == nullptr)
        return nullptr;

    TreeLinkNode *resu = nullptr;
    //所给的节点存在右子树的情况
    if (pNode->right != nullptr) {
        TreeLinkNode *tmpNode = pNode->right;
        while (tmpNode->left != nullptr) {
            tmpNode = tmpNode->left;
        }

        resu = tmpNode;
    }
    else {        //不存在右子树的情况
        TreeLinkNode *tmpNode = pNode;
        while (tmpNode->next != nullptr) {
            if (tmpNode == tmpNode->next->left) {
                resu = tmpNode->next;
                break;
            }
            tmpNode = tmpNode->next;
        }
    }

    return resu;
}