题解 | #二叉树的下一个结点#
二叉树的下一个结点
https://www.nowcoder.com/practice/9023a0c988684a53960365b889ceaf5e
基本思路:
- 如果有右子树,找右子树中的最左节点
- 如果没有右子树,向上遍历直到根节点或者为某个树的左子树,中序遍历下一个节点即为该左子树的根节点
class Solution { public: TreeLinkNode* GetNext(TreeLinkNode* pNode) { if (!pNode) return nullptr; TreeLinkNode* res; if (pNode->right) { res = pNode->right; while (res->left) res = res->left; return res; } res = pNode; while (res->next && res == res->next->right) res = res->next; return res->next; } };