AcWing 19. 二叉树的下一个节点
原题链接
中等
作者:
清_1
,
2021-04-11 17:10:10
,
所有人可见
,
阅读 262
算法1
思路:二叉树中给定节点后后序,分两种情况:
1:如果当前节点有右儿子,则右子树中最左侧的节点就是当前节点的后继
2:如果当前节点只有左儿子,则需沿着father一直向上找。找到第一个是其father左儿子的节点,该节点的father就是当前节点的后继
C++ 代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode *father;
* TreeNode(int x) : val(x), left(NULL), right(NULL), father(NULL) {}
* };
*/
class Solution {
public:
TreeNode* inorderSuccessor(TreeNode* p) {
//中序遍历找后继,一共分为两种情况,1:如果是右子树情况,后继要么是右子树的最左下角节点,要么只右儿子,返回即可
//第二这种情况是是左子树的后继,他是左子树中的右儿子,且不能继续向下找,那么他的后继就是father的father
if(p->right)//如果右子树存在
{
p=p->right;
while(p->left)p=p->left;
return p;
}
while(p->father && p==p->father->right) p=p->father;
return p->father;
}
};