TreeSet集合的特性:
1. 無序但可以排序，必須實現Comparable接口和重寫compareTo方法，簡單來說就是必須提供編譯器排序的依據方法，詳細可以看創建對象TreeSett的無參構造方法(Comparator = null)，以及新增元素使用add方法，底層調用TreeMap的put方法。
*不實現和重寫:
創建一個類，編譯器會報錯，錯誤類型: java.lang.ClassCastException
兩種解法:
(1) 實現Comparable接口和重寫 compareTo方法
(2) 使用Comparator比較器
(1) 第一種:
放在TreeSet集合中的元素需要實現 java.lang.Comparable接口，並且重寫compareTo方法
，compareTo方法需要編寫比較的邏輯 k.compareTo(t.key)，返回值可能是 0, <0, >0，
比較規則是程序員自己指定的，比如說升序或降序
import java.util.*;
public class test01 {
public static void main(String[] args) {
User u1 = new User("100", "Hina");
User u3 = new User("300", "Rui");
User u2 = new User("200", "Natsuo");
TreeSet<User> t = new TreeSet<>();
t.add(u1);
t.add(u2);
t.add(u3);
// 遍歷TreeSet集合
for(User u : t){
System.out.println(u);
}
}
}
// 寫一個User類 讓他實現Comparable接口
class User implements Comparable<User>{
private String idCard;
private String name;
public User(String idCard, String name){
this.idCard = idCard;
this.name = name;
}
// Setter and Getter
public String getIdCard() {
return idCard;
}
public void setIdCard(String idCard) {
this.idCard = idCard;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
// 重寫compareTo方法，這邊用
// String Integer.valueof()方法把String轉int讓他比較大小
@Override
public int compareTo(User u) {
int thisint = Integer.valueOf(this.getIdCard());
int userint = Integer.valueOf(u.idCard);
return thisint - userint;
}
// 重寫toString方法讓遍歷時顯示每個元素
public String toString(){
return "User[idCard= " + idCard + " , name= " + name + "]";
}
}
(2) 第二種: 構造TreeSet或TreeMap集合時傳一個比較器對象
import java.util.*;
public class test01 {
public static void main(String[] args) {
User u1 = new User("100", "Hina");
User u3 = new User("3000", "Rui");
User u2 = new User("20", "Natsuo");
// TreeSet<User> t= new TreeSet<>();
// 這樣不行，要傳遞比較器
// 給構造方法傳遞一個比較器
TreeSet<User> t = new TreeSet<>(new UserCompare());
t.add(u1);
t.add(u2);
t.add(u3);
// 遍歷TreeSet集合
for(User u : t){
System.out.println(u);
}
}
}
// 寫一個User類 讓他實現Comparable接口
class User{
private String idCard;
private String name;
public User(String idCard, String name){
this.idCard = idCard;
this.name = name;
}
// Setter and Getter
public String getIdCard() {
return idCard;
}
public void setIdCard(String idCard) {
this.idCard = idCard;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
// 重寫toString方法讓遍歷時顯示每個元素
public String toString(){
return "User[idCard= " + idCard + " , name= " + name + "]";
}
}
// 單獨寫一個比較器
// 比較器時現 java.util.Comparator接口
// Comparable是 java.lang包下
// Comparator是 java.util包下
class UserCompare implements Comparator<User>{
// 重寫Compare方法，制定比較條件
@Override
public int compare(User o1, User o2) {
int thisint = Integer.valueOf(o1.getIdCard());
int userint = Integer.valueOf(o2.getIdCard());
return thisint - userint;
}
}
或者使用匿名內部類:
不用單獨寫一個比較器，但只能單次使用，無法代碼復用
// 傳入比較器，比較器使用匿名內部類，直接new接口
TreeSet<User> t = new TreeSet<>(new Comparator<User>(){
public int compare(User o1, User o2) {
int thisint = Integer.valueOf(o1.getIdCard());
int userint = Integer.valueOf(o2.getIdCard());
return thisint - userint;
}
});
*比較兩種方法:
1. Comparable是 java.lang包下，Comparator是 java.util包下
2. Comparable接口實現後，把代碼寫進此類當中，Comparator接口要獨立寫一個類
3. 當比較規則不會發生變化或只有一個，實現Comparable接口
比較規則多個，需要多個比較規則之間切換，建議使用Comparator接口，且此接口的設計符合OCP原則。
