-
题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路:(1) 若该节点存在右子树:则下一个节点为右子树最左子节点(如图节点 B )
(2) 若该节点不存在右子树:这时分两种情况:
2.1 该节点为父节点的左子节点,则下一个节点为其父节点(如图节点 D )
2.2 该节点为父节点的右子节点,则沿着父节点向上遍历,知道找到一个节点的父节点的左子节点为该节点,则该节点的 父 节点下一个节点(如图节点 I ,沿着父节点一直向上查找找到 B ( B 为其父节点的左子节点),则 B 的父节点 A 为下 一个节点)。
-
/* 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==nullptr) return nullptr; TreeLinkNode* success; //该节点有右子树,所以该节点的下一个节点就是右子树最左边的那个节点 if(pNode->right != nullptr) { success=pNode->right; while(success->left !=nullptr) success=success->left; return success; } /* //该节点的无右子树,且该节点为该节点父节点的左子树,则该节点的下一个节点就是该节点的父节点 else if (pNode->next != nullptr && pNode->next->left==pNode) { return pNode->next; } //若该节点无右节点,且为该节点父节点的右节点,则往上找,直到找到一个 父节点1 为其父节点的左子树, //则该 父节点1 的父节点就为找到的节点 else if(pNode->next != nullptr && pNode->next->right==pNode) { while(pNode->next!= nullptr && pNode->next->left != pNode) pNode=pNode->next; return pNode->next; } //否则就没有下一个节点,返回的为空 else { return pNode->next; } */ //把没有右节点的情况整合下,使代码更加简洁 while(pNode->next!=nullptr&& pNode->next->left!=pNode) pNode=pNode->next; return pNode->next; } };