剑指 Offer（第2版）面试题 8：二叉树的下一个节点

剑指 Offer（第2版）面试题 8：二叉树的下一个节点

题目来源：19. 二叉树的下一个节点

解法1：分类讨论

在中序遍历的条件下：

• 如果一个节点有右子树，那么它的下一个节点就是它的右子树的最左侧子节点；
• 如果一个节点没有右子树：
  • 如果该节点是它父节点的左子节点，那么它的下一个节点就是它的父节点；
  • 如果该节点是它父节点的右子节点，沿着指向父节点的指针一直向上遍历，直到找到一个是它的父节点的左子节点的节点，那么这个节点的父节点就是我们要找的下一个节点。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode *father;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL), father(NULL) {}
 * };
 */
class Solution
{
public:
	TreeNode *inorderSuccessor(TreeNode *p)
	{
		if (p == nullptr)
			return nullptr;
		TreeNode *successor = nullptr;
		if (p->right)
		{
			TreeNode *pRight = p->right;
			while (pRight->left)
				pRight = pRight->left;
			successor = pRight;
		}
		else if (p->father)
		{
			if (p->father->left == p)
				successor = p->father;
			else
			{
				TreeNode *pCurrent = p, *pFather = p->father;
				while (pFather && pCurrent == pFather->right)
				{
					pCurrent = pFather;
					pFather = pFather->father;
				}
				successor = pFather;
			}
		}
		return successor;
	}
};

复杂度分析：

时间复杂度：O(Height)，其中 Height 是二叉树的高度。不论往上找还是往下找，总共遍历的节点数都不大于树的高度。

空间复杂度：O(1)。