分类讨论
1.有右孩子,则后继节点为右孩子的最左孩子。
2.如果没有右孩子,且当前节点是左孩子,则后继就是父节点。否则,当前节点就是右孩子,后继节点为一直往上走,直到该节点是左孩子为止,后继节点就是左孩子的父节点。
/*
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->right) {
p = p->right;
while(p->left) p = p->left;
return p;
}
while(p->next && p->next->right == p) p = p->next;
return p->next;
}
};