ACwing 19:二叉树的下一个节点
思路:
- 如果当前节点的右孩子节点不为空,那么找到当前节点的右孩子的最后一个不为空的左孩子节点,最后一个不为空的左孩子节点即为中序遍历序列的下一个节点
- 如果当前节点的右孩子节点为空,那么,从父节点一直想上找,判断当前的父节点是不是其父节点的左孩子节点,如果是的话该节点的父节点就是中序遍历的下一个节点
/**
* 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) {
// 如果当前节点有右孩子,那么右子树的最左侧节点就是当前节点的后继节点
// 如果当前节点没有右孩子节点,那么就需要沿着父节点向上寻找,找到第一个是其父节点左孩子节点的节点
// 那么该节点的父节点就是当前节点的后继节点。
if(p->right != nullptr)
{
p = p->right;
// 找到最左子节点
while(p->left != nullptr)
{
p = p->left;
}
return p;
}
// 如果节点的右孩子不存在 那么如果该节点是其父节点的左孩子,则返回父节点
// 那如果该节点并不是父节点的左孩子节点,那么就继续向上遍历,查找父节点的父节点 判断条件同上。
while(p->father != nullptr && p == p->father->right)
{
p = p->father;// 向上寻找父节点
// p = p->father->left 停止
}
return p->father;// 父节点就是下一个节点
}
};