题目:
给定一棵二叉搜索树和其中的一个节点 p
,找到该节点在树中的中序后继。如果节点没有中序后继,请返回 null
。
节点 p
的后继是值比 p.val
大的节点中键值最小的节点,即按中序遍历的顺序节点 p
的下一个节点
解题思路:
既然是查找某一节点的中序后继,所以我们使用中序深度优先搜索算法。定义全局变量(find = false;)用于记录是否查找到目标节点,当遍历到了我们想要的节点就将find置为true,并继续遍历下一个节点,该节点即为目标节点的中序后继节点,若后面没有其他节点,则返回空指针;若目标节点不存在,也返回空指针。具体代码实现如下:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
bool find = false;
TreeNode* res = nullptr;
void traverse(TreeNode* root, TreeNode* p) {
if (root == nullptr) return;
traverse(root->left, p);
if (res == nullptr && find == true) {
res = root;
return;
} else if (find == true && res != nullptr) {
return;
}
if (root == p) {
find = true;
}
traverse(root->right, p);
}
TreeNode* inorderSuccessor(TreeNode* root, TreeNode* p) {
traverse(root, p);
return res;
}
};