Java_基礎—TreeSet保證元素唯一和自然排序
阿新 • • 發佈:2019-01-23
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);
}
}