1题目
给定二叉搜索树(BST)的根节点 root 和一个整数值 val。
你需要在 BST 中找到节点值等于 val 的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 null 。
示例 1:
输入:root = [4,2,7,1,3], val = 2
输出:[2,1,3]
示例 2:
输入:root = [4,2,7,1,3], val = 5
输出:[]
2链接
题目链接:700. 二叉搜索树中的搜索 - 力扣(LeetCode)
视频链接:不愧是搜索树,这次搜索有方向了!| LeetCode:700.二叉搜索树中的搜索_哔哩哔哩_bilibili
3解题思路
二叉搜索树是一个有序树:
- 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值;
- 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值;
- 它的左、右子树也分别为二叉搜索树
这就决定了,二叉搜索树,递归遍历和迭代遍历和普通二叉树都不一样。
本题比较简单,不写过程了,直接看代码就懂了!
4代码
class Solution {
public:
//递归法
TreeNode* search(TreeNode* root, int val) {
if (root == nullptr || root->val == val) return root;
TreeNode* result = nullptr;
if (val > root->val) result = search(root->right, val);
else if (val < root->val) result = search(root->left, val);
return result;
}
// 迭代法
TreeNode* search1(TreeNode* root, int val) {
while (root != nullptr) {
if (val > root->val) root = root->right;
else if (val < root->val) root = root->left;
else return root;
}
return nullptr;
}
TreeNode* searchBST(TreeNode* root, int val) {
// return search(root, val);
return search1(root, val);
}
};