剑指Offer-8:二叉树的下一个节点

来源:互联网 发布:德国足球 知乎 编辑:程序博客网 时间:2024/06/05 03:27

题目:

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

问题解析:

寻找二叉树的中序遍历下一个节点,注意考虑不同的情况。

链接:

剑指Offer(第2版):P65

思路标签:

数据结构:二叉搜索树

指针实现。

解答:

1. C++

  • 分析树类的问题,最好绘制出二叉树的样例结构图,然后进行分析。
  • 注意题目中是要求寻找中序遍历的下一个节点。
  • 如果节点存在右子树:那么它的下一个节点就是右子树中的最左节点;
  • 如果节点没有右子树:
      1. 节点为其父节点的左子节点,那么其父节点就是它的下一个节点;
      1. 节点为其父节点的右子节点,那么需要沿其父指针一直向上遍历,一直找到某个节点是其父节点的左子节点为止,那么这个节点的父节点即是需要寻找的下一个节点。
/*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;        TreeLinkNode* pNext = nullptr;        if(pNode->right != nullptr){            TreeLinkNode* pRight = pNode->right;            while(pRight->left != nullptr)                pRight = pRight->left;            pNext = pRight;        }        else if(pNode->next != nullptr){            TreeLinkNode* pCurrent = pNode;            TreeLinkNode* pParent = pNode->next;            while(pParent != nullptr && pCurrent == pParent->right){                pCurrent = pParent;                pParent = pParent->next;            }            pNext = pParent;        }        return pNext;    }};
  • 细心,注意指针操作的正确性。
阅读全文
0 0
原创粉丝点击