如果你需要存储一些整数,并需要快速查找其中的最大值或最小值,那么可以将这些整数存储在一个二叉树中。在这个二叉树中,根节点的值为其中任意一个整数,每个左子节点都是小于等于其父节点的值,每个右子节点都是大于其父节点的值。
12
/ \
0 23
/ \
12 43
\
4242
当需要查找最大值时,只需要沿着右子节点一直往下搜索,直到找到没有右子节点的节点即为最大值节点;当需要查找最小值时,只需要沿着左子节点一直往下搜索,直到找到没有左子节点的节点即为最小值节点。这种搜索方式非常高效,时间复杂度为 O(log n)。
定义一个结点Node类,包含一个整数值以及左右子节点的引用;
public class Node {
int val;
Node left,right;
public Node(int val) {
this.val = val;
left = null;
right = null;
}
}
BinarySearchTree
类表示二叉搜索树,包含一个根节点的引用。
BinarySearchTree
类中定义了三个方法:
insert(int val)
:将一个整数值插入到二叉搜索树中。findMin()
:查找二叉搜索树中的最小值。findMax()
:查找二叉搜索树中的最大值。
public class BinarySearchTree {
Node root;
public BinarySearchTree() {
root = null;
}
public void insert(int val) {
root = insertHelper(root,val);
}
private Node insertHelper(Node node, int val) {
if (node==null)
return new Node(val);
if (val <= node.val)
node.left=insertHelper(node.left,val);
else {
node.right =insertHelper(node.right,val);
}
return node;
}
public int FindMin() {
if (root==null) {
throw new IllegalStateException("树为空");
}
Node curr = root;
while(curr.left!=null) {
curr=curr.left;
}
return curr.val;
}
public int FindMax() {
if (root==null) {
throw new IllegalStateException("树为空");
}
Node curr = root;
while(curr.right!=null) {
curr=curr.right;
}
return curr.val;
}
}
Main方法
首先,使用 Scanner
类从命令行读取一行字符串,保存到 input
变量中:
String input = scanner.nextLine();
然后,使用 String
类的 trim()
方法去除字符串两端的空白字符,并使用 split()
方法将字符串按照空白字符(空格、制表符、换行符等)分割成一个字符串数组:
String[] nums = input.trim().split("\\s+");
这里的 \\s+
是一个正则表达式,表示一个或多个空白字符。使用 split()
方法将字符串按照空白字符分割成数组后,每个字符串都代表一个整数。例如,如果用户输入了字符串 "1 2 3"
,则 nums
数组将包含三个字符串 "1"
、"2"
和 "3"
,分别代表整数 1、2 和 3。
最终,程序使用一个 for
循环遍历 nums
数组,将其中的每个字符串转换成整数,并依次插入到二叉搜索树中:
for (String num : nums) {
try {
int val = Integer.parseInt(num);
bst.insert(val);
} catch (NumberFormatException e) {
System.out.println("输入错误:" + num + " 不是整数");
}
}
如果某个字符串无法转换成整数,将抛出 NumberFormatException
异常,程序会捕获该异常并输出错误信息。
完整Main方法
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
BinarySearchTree bst = new BinarySearchTree();
System.out.print("请输入任意数量整数,以空格分隔:");
String input = sc.nextLine();
String[] nums = input.trim().split("\\s+");
for(String num : nums) //遍历数组插入二叉树
{
try {
int val = Integer.parseInt(num);
bst.insert(val);
}catch (NumberFormatException e) {
System.out.println("输入错误:"+num+"不是整数");
}
}
try{
int min = bst.FindMin();
System.out.println("最小值是:"+min);
}catch(IllegalStateException e) {
System.out.println("树为空,无法查找最小值");
}
try {
int max = bst.FindMax();
System.out.println("最大值是:"+max);
}catch(IllegalStateException e) {
System.out.println("树为空,无法查找最大值");
}
}
}
运行结果