【LeetCode & 剑指offer刷题】树题19：8 二叉树中序遍历的下一个结点
【LeetCode & 剑指offer 刷题笔记】目录（持续更新中...）
8 二叉树中序遍历的下一个结点
题目描述
给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。
分析：
结合图，我们可发现分成两大类：
(1) 有右子树的，那么下个结点就是
右子树最左边的点；（eg：D，B，E，A，C，G）
(2) 没有右子树的，是父节点左孩子（eg：N，I，L） ，那么
父节点就是下一个节点 ；
(3) 没有右子树的，是父节点的右孩子（eg：H，J，K，M）找他的父节点的父节点的父节点...直到遍历到的结点是其父节点的左孩子位置，
返回该父结点。如果没有eg：M，那么他就是尾节点。
/*
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
*
p
)
{
if
(
p
==
nullptr
)
return
nullptr
;
if
(
p
->
right
)
//
有右子树时，下个结点就是它右子树的最左子结点
{
p
=
p
->
right
;
while
(
p
->
left
!=
nullptr
)
p
=
p
->
left
;
return
p
;
}
else //没有右子树时，下个结点就是父结点（当前结点为左孩子时）或者向上遍历，以遍历到的结点为左孩子的父结点
{
while
(
p
->
next
)
{
if
(
p
->
next
->
left
==
p
)
//
若为父结点的左结点，则父结点就是下一个结点
return
p
->
next
;
else
p
=
p
->
next
;
//
往上查找，直到当前结点为父结点的左结点
}
return
nullptr
;
}
}
};
简洁版：
/**
*
后继：求二叉树中序遍历的下一个结点
方法：
(1)
有右子树的，那么下个结点就是右子树最左边的点；（在右边）
(2)
没有右子树的，是父结点的左孩子，返回父结点；（在右边）
(3)
没有右子树，是父结点的右孩子，从结点
x
开始向上查找，直到遍历到的结点是其父结点的左孩子位置，返回该父结点。（在右边）
*/
Node
*
BST_Successor
(
Node
*
node
)
{
if
(
node
->
right
)
//有右子树时，下个结点为右子树最左结点
{
Node
*
t
=
node
->
right
;
//右子树
while
(
t
->
left
)
t
=
t
->
left
;
return
t
;
}
else
//无右子树时，返回父结点(向上查找，结点为左孩子的父结点)
{
Node
*
par
=
node
->
parent
;
//
父结点
while
(
par
&&
par
->
right
==
node
)
//
如果当前结点是父结点的右孩子，向上遍历，直到当前结点为左结点
{
node
=
par
;
par
=
par
->
parent
;
}
//
退出时，当前结点为父结点的左孩子
return
par
;
//
返回父结点
}
}
posted @
2019-01-05 19:52 wikiwen 阅读(
...) 评论(
...) 编辑 收藏