關(guān)于TreeMap自定義排序規(guī)則的兩種方式

更新時間：2024年08月01日 09:58:58 作者：訾博ZiBo

這篇文章主要介紹了關(guān)于TreeMap自定義排序規(guī)則的兩種方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教

TreeMap自定義排序規(guī)則的兩種方式

1、概述

TreeMap基于二叉樹數(shù)據(jù)結(jié)構(gòu)存儲數(shù)據(jù)，同時實現(xiàn)了SortedMap接口以保障元素順序存取，默認(rèn)按照鍵的升序排序，也可以自定義排序比較器；

TreeMap常用于實現(xiàn)排序的映射列表，在使用TreeMap時其key必須實現(xiàn)Comparable接口或采用自定義的比較器，否則會拋出java.lang.ClassCastExption異常；

2、演示：其key實現(xiàn)Comparable接口

代碼演示：

package com.zibo.java.february.third;

import java.util.Set;
import java.util.TreeMap;

public class StuTreeMap {
    public static void main(String[] args) {
        TreeMap<Student,String> treeMap = new TreeMap<>();
        treeMap.put(new Student("大哥",24),"大哥");
        treeMap.put(new Student("二哥",23),"二哥");
        treeMap.put(new Student("三哥",22),"三哥");
        treeMap.put(new Student("四哥",21),"四哥");
        Set<Student> studentSet = treeMap.keySet();
        for (Student student : studentSet) {
            System.out.println(student.toString());
        }
    }
}
// 這里就省略getter和setter方法了
class Student implements Comparable<Student>{

    private String name;
    private int age;

    public Student() {
    }

    public Student(String name, int age) {
        this.name = name;
        this.age = age;
    }

    @Override
    public String toString() {
        return "Student{" +
                "name='" + name + '\'' +
                ", age=" + age +
                '}';
    }

    @Override
    public int compareTo(Student o) {
        // 實現(xiàn)Comparable接口在這里也是自定義排序規(guī)則
        // 如果什么也不寫，直接默認(rèn)return 0的話，只能存儲第一個被put的元素
        // 注意：升序就這么個寫法，不要看網(wǎng)上其他的什么相等返回0，相等的話要返回this.age，否則會出問題
        if(age > o.age){
            return 1;
        }else if(age < o.age){
            return -1;
        }
        return this.age;
    }
}

運行結(jié)果：

Student{name='四哥', age=21}
Student{name='三哥', age=22}
Student{name='二哥', age=23}
Student{name='大哥', age=24}

3、演示：采用自定義比較器

代碼演示：

package com.zibo.java.february.third;

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

public class StuTreeMap2 {
    public static void main(String[] args) {
        TreeMap<Student2,String> treeMap = new TreeMap<>(new Comparator<Student2>() {
            @Override
            public int compare(Student2 o1, Student2 o2) {
                // 基本和key實現(xiàn)Comparable接口，重寫compareTo方法一致
                // 升序排序就是這么寫的
                if(o1.getAge() > o2.getAge()){
                    return 1;
                }else if(o1.getAge() < o2.getAge()){
                    return -1;
                }
                // 相等的時候不能返回0
                return o1.getAge();
            }
        });
        treeMap.put(new Student2("大哥",24),"大哥");
        treeMap.put(new Student2("二哥",23),"二哥");
        treeMap.put(new Student2("三哥",22),"三哥");
        treeMap.put(new Student2("四哥",21),"四哥");
        Set<Student2> studentSet = treeMap.keySet();
        for (Student2 student : studentSet) {
            System.out.println(student.toString());
        }
    }
}

class Student2{

    private String name;
    private int age;

    public Student2() {
    }

    public Student2(String name, int age) {
        this.name = name;
        this.age = age;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    @Override
    public String toString() {
        return "Student{" +
                "name='" + name + '\'' +
                ", age=" + age +
                '}';
    }
}

運行結(jié)果：