package cn.anson.tree;

import java.util.*;

/**
* 定義一個TreeSet物件，儲存自定義物件Student。 按照姓名長度的大小決定儲存的順序，從長到短排序，如果長度一樣，年齡小的在前面
*/

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

public Student(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;
}

// 姓名的長度
public int compareTo(Student s) {
int num = this.name.length() - s.name.length();
// 很多時候，別人給我們的需求其實只是一個主要需要
// 還有很多的次要需求是需要我們自己進行分析的。
// 比較姓名的內容
int num2 = (num == 0) ? (this.name.compareTo(s.name)) : num;
// 繼續分析，姓名長度和內容都相同的情況下，年齡還可能不一樣呢?
// 所以，當姓名長度和內容都相同的時候，我們在比較下年齡就好了
int num3 = (num2 == 0) ? (this.age - s.age) : num2;
return num3;
}
}

public class treeSetDemo {
public static void main(String[] args) {
TreeSet<Student> ts = new TreeSet<Student>(new Comparator<Student>() {

public int compare(Student s1, Student s2) {
// 按年齡排序，從小到大
int num = s1.getAge() - s2.getAge();
// 次要條件
int num2 = (num == 0) ? (s1.getName().compareTo(s2.getName()))
: num;
return num2;
}
});

// 建立元素物件
Student s1 = new Student("小李", 52);
Student s2 = new Student("小東", 60);
Student s3 = new Student("小紅", 44);
Student s4 = new Student("小明", 34);

// 新增元素
ts.add(s1);
ts.add(s2);
ts.add(s3);
ts.add(s4);
// 遍歷
for (Student s : ts) {
System.out.println(s.getName() + "***" + s.getAge());
}
}
}