怎么在Java中對(duì)TreeSet進(jìn)行自定義類(lèi)型的排序

怎么在Java中對(duì)TreeSet進(jìn)行自定義類(lèi)型的排序？相信很多沒(méi)有經(jīng)驗(yàn)的人對(duì)此束手無(wú)策，為此本文總結(jié)了問(wèn)題出現(xiàn)的原因和解決方法，通過(guò)這篇文章希望你能解決這個(gè)問(wèn)題。

Java可以用來(lái)干什么

Java主要應(yīng)用于：1. web開(kāi)發(fā)；2. Android開(kāi)發(fā)；3. 客戶端開(kāi)發(fā)；4. 網(wǎng)頁(yè)開(kāi)發(fā)；5. 企業(yè)級(jí)應(yīng)用開(kāi)發(fā)；6. Java大數(shù)據(jù)開(kāi)發(fā)；7.游戲開(kāi)發(fā)等。

TreeSet與Java自定義類(lèi)型的排序

• 演示TreeSet對(duì)String是可排序的

• TreeSet無(wú)法對(duì)自定義類(lèi)型進(jìn)行排序

• 比較規(guī)則怎么寫(xiě)

• 自平衡二叉樹(shù)結(jié)構(gòu)

• 實(shí)現(xiàn)比較器接口

• Collections工具類(lèi)

演示TreeSet對(duì)String是可排序的

1.TreeMap集合底層實(shí)際上是一個(gè)TreeMap
2.TreeMap集合底層是一個(gè)二叉樹(shù)
3.放到TreeSet集合中的元素，等同于放到TreeMap集合key部分了
4.TreeSet集合中的元素，無(wú)序不可重復(fù)，但是可以按照元素的大小順序自動(dòng)排序
稱(chēng)為：可排序集合
例如：編寫(xiě)程序從數(shù)據(jù)庫(kù)中取出數(shù)據(jù)，在頁(yè)面展示用戶信息的時(shí)候按照生日升序或者降序，
這個(gè)時(shí)候可以使用TreeSet集合，因?yàn)門(mén)reeSet集合放進(jìn)去，拿出來(lái)就是有序的。

//創(chuàng)建一個(gè)TreeSet集合
  TreeSet<String> ts=new TreeSet<>();
  //添加Stringts.add("zhangsan");ts.add("lisi");ts.add("wangwu");ts.add("zhangsi");ts.add("wangliu");for(String s:ts){
      //按照字典順序排序
      System.out.print(s+" ");
  }
  TreeSet<Integer> ts2=new TreeSet<>();ts2.add(100);ts2.add(200);ts2.add(900);ts2.add(800);  ts2.add(600);ts2.add(10);for(Integer i:ts2){
      //按照升序排序
      System.out.print(i+" ");}

TreeSet無(wú)法對(duì)自定義類(lèi)型進(jìn)行排序

TreeSet可以對(duì)自定義類(lèi)型進(jìn)行排序？
以下程序中，對(duì)于Person類(lèi)來(lái)說(shuō)，無(wú)法排序，因?yàn)闆](méi)有指定Person對(duì)象之間的比較規(guī)則。誰(shuí)大誰(shuí)小并沒(méi)有說(shuō)明。

public class TreeSetTest02 {
    public static void main(String[] args) {
        Person p1=new Person(50);
        Person p2=new Person(10);
        Person p3=new Person(20);
        Person p4=new Person(60);
        Person p5=new Person(40);
        Person p6=new Person(30);
        TreeSet<Person> persons=new TreeSet<>();
        persons.add(p1);
        persons.add(p2);
        persons.add(p3);
        persons.add(p4);
        persons.add(p5);
        persons.add(p6);
        for(Person p:persons){
            System.out.println(p);
        }
    }}class Person{
    int age;
    public Person(int age){
        this.age=age;
    }
    @Override
    public String toString() {
        return "Person [age=" + age + "]";
    }}

Exception in thread "main" java.lang.ClassCastException: testCollection.Person cannot be cast to java.lang.Comparable

出現(xiàn)這個(gè)錯(cuò)誤的原因是
Person類(lèi)沒(méi)有實(shí)現(xiàn)java.lang,Comparable接口

//放在TreeSet集合中的元素需要實(shí)現(xiàn)java.lang.Comparable接口
//并且實(shí)現(xiàn)compareTo方法，equals可以不寫(xiě)

  public class TreeSetTest04 {
        public static void main(String[] args) {
            Customer p1=new Customer(50);
            Customer p2=new Customer(10);
            Customer p3=new Customer(20);
            Customer p4=new Customer(60);
            Customer p5=new Customer(40);
            Customer p6=new Customer(30);
            TreeSet<Customer> customers=new TreeSet<>();
            customers.add(p1);
            customers.add(p2);
            customers.add(p3);
            customers.add(p4);
            customers.add(p5);
            customers.add(p6);
            for(Customer p:customers){
                System.out.println(p);
            }
        }
    }
    //放在TreeSet集合中的元素需要實(shí)現(xiàn)java.lang.Comparable接口//并且實(shí)現(xiàn)compareTo方法，equals可以不寫(xiě)
    class Customer implements Comparable<Customer>{
        int age;
        public Customer(int age){
            this.age=age;
        }
        @Override
        public String toString() {
            return "Customer [age=" + age + "]";
        }
        //需要在這個(gè)方法中編寫(xiě)比較的邏輯，或者說(shuō)比較的規(guī)則，按照什么進(jìn)行比較。
        //k.compareTo(t.key)
        //拿著參數(shù)k和集合中的每個(gè)k進(jìn)行比較，返回值可能是>0,<0,=0
        //比較規(guī)則最終還是程序員實(shí)現(xiàn)的：例如按照年齡升序，或者按照年齡降序
        @Override
        public int compareTo(Customer c) {    //c1.compareTo(c2)// TODO Auto-generated method stub
            //this是c1
            //c是c2
            //c1和c2進(jìn)行比較的時(shí)候，就是this和c比較//    int age1=this.age;//    int age2=c.age;//    if(age1==age2){//       return 0;//    }else if(age1>age2){//       return 1;//    }else{//       return -1;//    }
            return this.age-c.age;    //>,<,=
        }

    }

//需要在這個(gè)方法中編寫(xiě)比較的邏輯，或者說(shuō)比較的規(guī)則，按照什么進(jìn)行比較。
//k.compareTo(t.key)
//拿著參數(shù)k和集合中的每個(gè)k進(jìn)行比較，返回值可能是>0,<0,=0
//比較規(guī)則最終還是程序員實(shí)現(xiàn)的：例如按照年齡升序，或者按照年齡降序

比較規(guī)則怎么寫(xiě)

先按照年齡升序，如果年齡一樣的再按照姓名升序

public class TreeSetTest05 {
    public static void main(String[] args) {
        TreeSet<Vip> vips=new TreeSet<>();
        vips.add(new Vip("zhangsi",20));
        vips.add(new Vip("zhangsan",20));
        vips.add(new Vip("king",18));
        vips.add(new Vip("soft",17));
        for(Vip vip:vips){
            System.out.println(vip);
        }
    }}class Vip implements Comparable<Vip>{
    String name;
    int age;
    public Vip(String name,int age){
        this.name=name;
        this.age=age;
    }
    @Override
    public String toString() {
        return "Vip [name=" + name + ", age=" + age + "]";
    }
    //compareTo方法的返回值很重要：
    //返回0表示相同，value會(huì)覆蓋
    //>0,會(huì)繼續(xù)在右子樹(shù)上找
    //<0，會(huì)繼續(xù)在左子樹(shù)上找

    @Override
    public int compareTo(Vip v) {
        if(this.age==v.age){
            //年齡相同時(shí)，按照名字排序
            //姓名是String類(lèi)型，可以直接比，調(diào)用compareTo方法
            return this.name.compareTo(v.name);
        }else{
            //年齡不一樣
            return this.age-v.age;
        }
    }}

自平衡二叉樹(shù)結(jié)構(gòu)

1.自平衡二叉樹(shù)，遵循左小右大的原則存放
2.遍歷二叉樹(shù)的時(shí)候有三種方式
前序遍歷：根左右
中序遍歷：左根右
后序遍歷：左右根
注意：前中后說(shuō)的是根的位置
3.TreeSet集合和TreeMap集合采用的是中序遍歷方式，即左根右。他們是自平衡二叉樹(shù)
100 200 50 60 80 120 140 130 135 180 666

實(shí)現(xiàn)比較器接口

TreeSet集合中元素可排序的第二種方式，使用比較器的方式

public class TreeSetTest06 {
    public static void main(String[] args) {
        //創(chuàng)建TreeSet集合的時(shí)候，需要使用比較器
        //TreeSet<Wugui> wuGuis=new TreeSet<>();   //這樣不行，沒(méi)有通過(guò)構(gòu)造方法傳遞一個(gè)比較器進(jìn)去
        TreeSet<Wugui> wuGuis=new TreeSet<>(new WuguiComparator());
        wuGuis.add(new Wugui(1000));
        wuGuis.add(new Wugui(800));
        wuGuis.add(new Wugui(900));
        wuGuis.add(new Wugui(300));
        wuGuis.add(new Wugui(60));
        for(Wugui wugui:wuGuis){
            System.out.println(wugui);
        }

    }}class Wugui{
    int age;

    public Wugui(int age) {
        super();
        this.age = age;
    }

    @Override
    public String toString() {
        return "Wugui [age=" + age + "]";
    }}//單獨(dú)再這里編寫(xiě)一個(gè)比較器//比較器實(shí)現(xiàn)java.util.Comparator接口（Comparable是java.lang包下的）class WuguiComparator implements Comparator<Wugui>{
    public int compare(Wugui o1,Wugui o2){
        //指定比較規(guī)則
        //按照年齡排序
        return o1.age-o2.age;
    }}

我們可以使用匿名內(nèi)部類(lèi)的方式
可以使用匿名內(nèi)部類(lèi)的方式（這個(gè)類(lèi)沒(méi)有名字，直接new接口）

TreeSet<Wugui> wuGuis=new TreeSet<>(new Comparator<Wugui>(){public int compare(Wugui o1,Wugui o2){
        //指定比較規(guī)則
        //按照年齡排序
        return o1.age-o2.age;
        }});

最終的結(jié)論，放在TreeSet或者TreeMap集合key部分的元素要想做到排序，包括兩種方式
第一種：放在集合中的元素實(shí)現(xiàn)java.lang.Comparable接口
第二種：在構(gòu)造TreeSet或者TreeMap集合的時(shí)候給它傳一個(gè)比較器對(duì)象。
Comparable和Comparator怎么選擇呢？
當(dāng)比較規(guī)則不會(huì)發(fā)生改變的時(shí)候，或者說(shuō)當(dāng)比較規(guī)則只有1個(gè)的時(shí)候，建議實(shí)現(xiàn)Comparable接口
如果比較規(guī)則有多個(gè)，并且需要多個(gè)比較規(guī)則之間頻繁切換，建議使用comparator接口
comparator接口的設(shè)計(jì)符合OCP原則。

Collections工具類(lèi)

java.util.Collections集合工具類(lèi)，方便集合的操作

public class CollectionsTest {
    static class Wugui2 implements Comparable<Wugui2>{
        int age;

        public Wugui2(int age) {
            super();
            this.age = age;
        }

        @Override
        public String toString() {
            return "Wugui2 [age=" + age + "]";
        }

        @Override
        public int compareTo(Wugui2 o) {
            // TODO Auto-generated method stub
            return this.age-o.age;
        }
    }
    public static void main(String[] args) {
        //ArrayList集合不是線程安全的
        List<String> list=new ArrayList<String>();
        //變成線程安全的
        Collections.synchronizedList(list);
        //排序
        list.add("abc");
        list.add("abe");
        list.add("abd");
        list.add("abf");
        list.add("abn");
        list.add("abm");
        Collections.sort(list);
        for(String s:list){
            System.out.println(s);
        }
        List<Wugui2> wuguis=new ArrayList<>();
        wuguis.add(new Wugui2(1000));
        wuguis.add(new Wugui2(8000));
        wuguis.add(new Wugui2(4000));
        wuguis.add(new Wugui2(6000));
        //注意：對(duì)list集合中元素排序，需要保證list集合中元素實(shí)現(xiàn)了Comparable接口
        Collections.sort(wuguis);
        for(Wugui2 wugui:wuguis){
            System.out.println(wugui);
        }
        //對(duì)set集合怎么排序呢
        Set<String> set=new HashSet<>();
        set.add("king");
        set.add("kingsoft");
        set.add("king2");
        set.add("king1");
        //將set集合轉(zhuǎn)換成list集合
        List<String> myList=new ArrayList<>(set);
        Collections.sort(myList);
        for(String s:myList){
            System.out.println(s);
        }
        //這種方式也可以排序
        //Collections.sort(list集合，比較器對(duì)象)
    }}

看完上述內(nèi)容，你們掌握怎么在Java中對(duì)TreeSet進(jìn)行自定義類(lèi)型的排序的方法了嗎？如果還想學(xué)到更多技能或想了解更多相關(guān)內(nèi)容，歡迎關(guān)注億速云行業(yè)資訊頻道，感謝各位的閱讀！