题目:
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路:
考虑中序遍历的过程,
如果当前结点存在右子节点,那么当前结点的下一个结点应该为该右子树的最左边的结点;
如果当前结点没有右子节点,那么如果当前结点的父结点p->parent为其父结点的父结点p->parent->patent的左子结点,那么当前结点的下一结点即为其父结点;否则继续往上遍历,直至找到一个结点为其父结点的左子结点,那么该结点即为当前结点的下一结点,如果找不到,则返回NULL;
(文字描述比较晦涩,大家可以画图看看)
在线测试OJ:
http://www.nowcoder.com/books/coding-interviews/9023a0c988684a53960365b889ceaf5e?rp=3
AC代码:
using namespace std;
/*
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;
if(pNode->right!=NULL){
TreeLinkNode* pTmp=pNode->right;
while(pTmp->left!=NULL){
pTmp=pTmp->left;
}
return pTmp;
}
else{
while(pNode->next!=NULL){
if(pNode==pNode->next->left)
return pNode->next;
pNode=pNode->next;
}
return NULL;
}
}
};