56二叉树的下一个结点

题目描述

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

思路：

这题还是自己的思路错误，还有就是对二叉树的遍历没有理解，只是记得左中右，根的前一个节点是左子树遍历的最后一个节点，根节点的下一个节点是右子树遍历的第一个节点，即右子树最左边的节点。

本题的思路是：按照当前节点是否有右子树进行划分。

如果有右子树，那么就是右子树遍历到的最左边的孩子节点。

如果没有右子树，就需要向上访问父节点，只要该父节点的父节点左孩子是父节点，就是第一个左转的节点，即为答案。

自己在举例子的时候是平衡二叉树，结论是父节点的父节点为所求，这种特殊情况结论是不对的。

还有就是指针也可以直接比较是否相等。

//分析二叉树的下一个节点，一共有以下情况：
//1.二叉树为空，则返回空；
//2.节点右孩子存在，则设置一个指针从该节点的右孩子出发，一直沿着指向左子结点的指针找到的叶子节点即为下一个节点；
//3.节点不是根节点。如果该节点是其父节点的左孩子，则返回父节点；否则继续向上遍历其父节点的父节点，重复之前的判断，返回结果。

/*
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;
        }
        //have right subtree
        if(pNode->right != nullptr){
            TreeLinkNode* nextRight = pNode -> right;
            while(nextRight -> left != nullptr){
                nextRight = nextRight->left;
            }
            return nextRight;
        }
        //have not right subtree
        else{
            while(pNode->next){
                if((pNode->next->left != nullptr) && (pNode->next->left->val == pNode->val)){//可以直接比较指针
                    return pNode->next;
                }
                pNode = pNode->next;
            }
        }
        return nullptr;
    }
};