您的位置:首页 > 理论基础 > 数据结构算法

【剑指offer】数据结构 —— 二叉树的下一个结点(C++)

2018-04-01 13:24 405 查看
题目描述

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

思路讲解

考虑到中序遍历中,结点的相对顺序是左根右,所以,一个结点中序遍历顺序的下一个结点, 先顺着右结点找,没有就顺着根结点往上找

第一种:结点p存在右结点,那么下一个结点就是p的右结点的最左子节点

第二种:结点p不存在右结点,那么下一个结点就是p的根节点序列中,第一个使得p在其左分支内的(从左根右的顺序理解)

代码实现(牛客网版本)

/*
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 == NULL) return NULL;
//第一种情况,找右结点中最左子节点
if(pNode -> right != NULL){
TreeLinkNode* p = pNode -> right;
while(p -> left != NULL)
p = p -> left;
return p;
}
//第二种情况,找根序列中,第一个p在其左分支中的
else{
TreeLinkNode* p = pNode;
while(p -> next != NULL && p -> next ->left != p) p = p -> next;
return p -> next;
}
}
};


拓展引申

此题考查的是中序遍历的顺序
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  数据结构 二叉树