问题描述:
给定一棵二叉树的其中一个节点,请找出中序遍历序列的下一个节点。注意: 如果给定的节点是中序遍历序列的最后一个,则返回空节点;二叉树一定不为空,且给定的节点一定不是空节点;
样例
求解思路,分情况讨论,第一种情况,比如F节点,它的下一个节点为右子树里面最左边的节点;第二种情况,比如D节点,它的下一个节点是F点,D节点的父节点是C,D是C的右节点,且C是F节点的左节点。while(p->father&&p==p->father->right) 当不满足这个情况的时候跳出循环,也就是当p= =p->father->left的时候,条件不满足,那么P返回的是C节点,最后return 的是F节点,相反,G节点就不满足,因为循环一遍后,P的值为F节点,那么return掉的是F节点的父节点,为空NULL。第三种情况,A节点,左右都不存在那么返回其父节点即可。G节点返回的是NULL。
if(p->right)可以讨论如果右边存在且左边不存在情况,如果while循环无法执行,返回的是其右节点。也可以判断右边存在,左边不存在的情况,
while()j加上return可以判断,左边存在,右边不存在的情况,也可以判断左右均不存在的情况。
class Solution {
public:
TreeNode* inorderSuccessor(TreeNode* p) {
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;
}
};