71 lines
1.7 KiB
Java
71 lines
1.7 KiB
Java
//给定二叉搜索树（BST）的根节点和一个值。 你需要在BST中找到节点值等于给定值的节点。 返回以该节点为根的子树。 如果节点不存在，则返回 NULL。
|
||
//
|
||
// 例如，
|
||
//
|
||
//
|
||
//给定二叉搜索树:
|
||
//
|
||
// 4
|
||
// / \
|
||
// 2 7
|
||
// / \
|
||
// 1 3
|
||
//
|
||
//和值: 2
|
||
//
|
||
//
|
||
// 你应该返回如下子树:
|
||
//
|
||
//
|
||
// 2
|
||
// / \
|
||
// 1 3
|
||
//
|
||
//
|
||
// 在上述示例中，如果要找的值是 5，但因为没有节点值为 5，我们应该返回 NULL。
|
||
// Related Topics 树 二叉搜索树 二叉树 👍 162 👎 0
|
||
|
||
package leetcode.editor.cn;
|
||
|
||
import com.code.leet.entiy.TreeNode;
|
||
|
||
//700:二叉搜索树中的搜索
|
||
class SearchInABinarySearchTree {
|
||
public static void main(String[] args) {
|
||
//测试代码
|
||
Solution solution = new SearchInABinarySearchTree().new Solution();
|
||
}
|
||
|
||
//力扣代码
|
||
//leetcode submit region begin(Prohibit modification and deletion)
|
||
|
||
/**
|
||
* Definition for a binary tree node.
|
||
* public class TreeNode {
|
||
* int val;
|
||
* TreeNode left;
|
||
* TreeNode right;
|
||
* TreeNode() {}
|
||
* TreeNode(int val) { this.val = val; }
|
||
* TreeNode(int val, TreeNode left, TreeNode right) {
|
||
* this.val = val;
|
||
* this.left = left;
|
||
* this.right = right;
|
||
* }
|
||
* }
|
||
*/
|
||
class Solution {
|
||
public TreeNode searchBST(TreeNode root, int val) {
|
||
while (root != null && root.val != val) {
|
||
if (root.val > val) {
|
||
root = root.left;
|
||
} else {
|
||
root = root.right;
|
||
}
|
||
}
|
||
return root;
|
||
}
|
||
}
|
||
//leetcode submit region end(Prohibit modification and deletion)
|
||
|
||
} |