題目描述
思路:兩類情況,一是正常的左根右順序訪問,另外一種是,沒有左孩子,下一節點將會指向上一節點的根節點,或者最後爲空的情況。
/*
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* pNode)
{
if(pNode==NULL) return NULL;
TreeLinkNode* pNext=NULL;
if(pNode->right)//有右子樹,找到右子樹的最左的結點
{
pNode=pNode->right;
while(pNode->left) pNode=pNode->left;
pNext=pNode;
}
else if(pNode->next)
{
TreeLinkNode* pParent=pNode->next;
while(pParent && pNode==pParent->right)
{
pNode=pParent;
pParent=pParent->next;
}
pNext=pParent;
}
return pNext;
}
};