学习历程-二叉树的下一个节点

题目描述

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

思路

对于中序遍历遍历顺序为 左中右
1.如果这个节点有右孩子，那么根据顺序。可以得出下一个点为右子树最“左侧”的那个点，

2.如果这个节点没有右孩子，这代表这个点所代表的子树已经遍历完了，需要回到其父亲那里看看情况
2.1.没有右孩子，而且没父亲，那这个节点是根节点，这个节点遍历完了代表整颗树都遍历完了，返回null，
2.2.没有右孩子，而且是父亲的左孩子，根据顺序，可以得出下一个节点就是其父亲,例如点"D" “F”
2.3.没有右孩子，而且是其父亲节点的右孩子，根据顺序，代表其父亲节点遍历完了，我们需要看看其父亲节点的情况。（一直向上找父亲，看看每个“父亲”的情况）例如点“D”“E”“F”“G”
2.3.0.这个点为其父亲右孩子。根据顺序，代表这个点的父亲所代表的树遍历完了，我们需要再看看其父亲的父亲。例如点“C”
2.3.1.找到一个为其父亲节点的左孩子的点，代表直到这个节点为止都遍历完了，下个节点为这个点的父亲。例如点“B”
2.3.2.父亲为空，整个树都完毕了;例如点“A”

/*
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:
    //中序遍历任意节点 返回其 中序遍历的下一节点
    //情况1，有右孩子，返回右孩子的最左节点（节点没有左孩子了）
    //2，没有有孩子
    //2.1，其为根节点（父节点为空），返回NULL
    //2.2，不是根节点，
    //2.2.1，其为父节点的右孩子，那么代表其父节点树遍历完毕，即寻找其父节点的的父节点，直到其不是右孩子，如果循环至根节点，返回NULL。代表遍历完毕了
    //2.2.2，其为父节点的左孩子，返回其父节点
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if (pNode == NULL)
            return NULL;
        if (pNode->right != NULL)
        {
            //有右孩子
            TreeLinkNode* temp=pNode->right;
            while (temp->left!=NULL)
            {
                temp = temp->left;
            }
            return temp;
        }
        else
        {
            //没有有孩子

            //其为根节点
            if (pNode->next == NULL)
            {
                return NULL;
            }
            else
            {
                //不是根节点
                if (pNode->next->left==pNode)
                {
                    //其为父节点的左孩子
                    return pNode->next;
                }
                else
                {
                    //其为父节点的右孩子
                    TreeLinkNode* temp = pNode->next;
                    while (1)
                    {
                        //父^n节点是根节点
                        if (temp->next == NULL)
                            return NULL;
                        //父^n节点 是 父^n+1节点 的右节点
                        if (temp->next->right == temp)
                        {
                            temp = temp->next;
                        }
                        else
                        {
                            return temp->next;
                        }
                    }
                    return temp;
                }
            }
        }
    }
};