1. 程式人生 > >Java_基礎—TreeSet保證元素唯一和自然排序

Java_基礎—TreeSet保證元素唯一和自然排序

TreeSet儲存自定義物件時,必須使該物件實現Comparable介面,並重寫該方法。

public class Person implements Comparable<Person>{

@Override
    public int compareTo(Person o) {    
        //return 0;      //返回0,集合中只有一個元素
        return 1;        //返回1,集合怎麼存就怎麼取
        //return -1;     //返回-1,集合會將儲存的元素倒敘
    }
}

二叉樹:小的儲存在左邊(負數),大的儲存在右邊(正數),相等就不存(0)

compareTo方法,在TreeSet集合如何儲存元素取決於 compareTo方法的返回值

  • 返回0,集合中只有一個元素
  • 返回-1,集合會將儲存的元素倒敘
  • 返回1,集合怎麼存就怎麼取

自定義一個Person類,屬性為name, age

package com.soar.bean;

public class Person implements Comparable<Person>{
    private String name;
    private int age;


    public Person() {
        super();    
    }
    public
Person(String name, int age) { super(); 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; } public String toString() { return "Person [name=" + name + ", age=" + age + "]"; } //Alt + Shift + S → Generate hashCode() and equals()方法 /* * final int prime = 31; * 為什麼是31? * ①31是個質數,質數是能被1和自己本身整除的數 * ②31這個數既不大也不小 * ③31這個數好算,2的五次方-1,2向左移動5位 */ @Override public int hashCode() { final int prime = 31; int result = 1; result = prime * result + age; result = prime * result + ((name == null) ? 0 : name.hashCode()); return result; } @Override public boolean equals(Object obj) { if (this == obj) //呼叫的物件和傳入的物件是同一個物件 return true; //直接返回true if (obj == null) //傳入的物件為null return false; //返回false if (getClass() != obj.getClass()) //判斷兩個物件對應的位元組碼檔案是否是同一個 return false; //如果不是,直接返回false Person other = (Person) obj; //向下轉型 if (age != other.age) //呼叫物件的年齡不等於傳入物件的年齡 return false; //返回false if (name == null) { //呼叫物件的姓名為null if (other.name != null) //傳入物件的姓名不為null return false; //返回flase } else if (!name.equals(other.name)) //呼叫的姓名不等於傳入物件的姓名 return false; //返回false return true; //返回true } @Override public int compareTo(Person o) { int num = this.age - o.age; //年齡是比較的主要條件 return num == 0 ? this.name.compareTo(o.name) : num; //姓名是比較的次要條件 } }

定義一個TreeSet類

package com.soar.set;

import java.util.TreeSet;

import com.soar.bean.Person;

public class Demo3_TreeSet {
    /*
     *A:案例演示
     *  TreeSet儲存Integer型別的元素並遍歷
     *  TreeSet集合是用來對元素進行排序的,同樣他也可以保證元素的唯一
     *  當compareTo方法返回0的時候,集合中只有一個一個元素
     *  當compareTo方法返回正數的時候,集合會怎麼存就怎麼取
     *  當compareTo方法返回負數的時候,集合會倒序儲存
     */
    public static void main(String[] args) {
        //add();
        TreeSet<Person> ts = new TreeSet<>();
        ts.add(new Person("張三",23));
        ts.add(new Person("李四",24));
        ts.add(new Person("王五",25));
        ts.add(new Person("趙六",26));
        ts.add(new Person("周七",26));

        System.out.println(ts); 
    }

    private static void add() {
        TreeSet<Integer> ts = new TreeSet<>();
        ts.add(1);
        ts.add(2);
        ts.add(2);
        ts.add(3);
        ts.add(3);

        System.out.println(ts);
    }

}