Java中的TreeMap集合

最新推荐文章于 2022-04-19 12:12:10 发布

原创最新推荐文章于 2022-04-19 12:12:10 发布 · 334 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#TreeMap #Comparabale接口

Java 专栏收录该内容

63 篇文章

订阅专栏

本文介绍了如何使用TreeMap并自定义元素排序规则。通过实现Comparator接口，可以按照自定义的规则（如学生姓名和年龄）对元素进行排序，示例代码展示了具体的实现过程。

TreeMap中的元素默认按照keys的自然排序排列。（对Integer来说，其自然排序就是数字的升序；对String来说，其自然排序就是按照字母表排序）

如果想实现让元素按自己的想法去进行排序，可以实现Comparable接口之后重写compare接口，在里面书写自己的规则，之后就可以实现自己的功能。

代码：

import java.util.Comparator;
import java.util.TreeMap;

import com.heima.bean.Student;

public class Demo6_TreeMap {
	public static void main(String[] args) {
		//demo1();
		TreeMap<Student, String> tm = new TreeMap<>(new Comparator<Student>() {

			@Override
			public int compare(Student s1, Student s2) {
				int num = s1.getName().compareTo(s2.getName());
				return num == 0 ? s1.getAge() - s2.getAge() : num;
			}
		});
		tm.put(new Student("张三", 23), "北京");
		tm.put(new Student("赵六", 26), "上海");			
		tm.put(new Student("李四", 24), "广州");
		tm.put(new Student("王五", 25), "深圳");
		System.out.println(tm);
	}

	private static void demo1() {
		TreeMap<Student, String> tm = new TreeMap<>();
		tm.put(new Student("张三", 23), "北京");
		tm.put(new Student("赵六", 26), "上海");			
		tm.put(new Student("李四", 24), "广州");
		tm.put(new Student("王五", 25), "深圳");
		System.out.println(tm);
	}
}