java TreeSet

TreeSet可以對Set集合中的元素進行排序。

底層數據結構：二叉樹。

保證元素唯一：comoareTo  return 0；

TreeSet排序的第一種方式：讓元素自身具備比較性。元素實現Comparable介面覆蓋compareTo方法，這種方式也稱為元素的自然順序，或者叫預設順序。

TreeSet排序的第二種方式：當元素自身不具備比較性時，或者具備的比較性不是所需要的，這時需要讓集合自身具備比較性。在集合初始化時，就有了比較方式。

存儲自定義對象學生。

記住：排序時，當主要條件相同時，一定要判斷一下次要條件。

public class TreeSetDemo {
    public static void main(String[] args) {
        TreeSet ts = new TreeSet();
        ts.add(new Student("lisi02", 2));
        ts.add(new Student("lisi01", 1));
        ts.add(new Student("lisi04", 4));
        ts.add(new Student("lisi03", 3));
        ts.add(new Student("lisi04", 3));
        Iterator it = ts.iterator();
        while (it.hasNext()) {
            Student s = (Student) it.next();
            System.out.println(s.getName() + "::" + s.getAge());
        }
    }
}

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

    public Student(String name, int age) {
        this.name = name;
        this.age = age;
    }

    public String getName() {
        return name;
    }

    public int getAge() {
        return age;
    }

    @Override
    public int compareTo(Object o) {
        if (!(o instanceof Student))
            throw new RuntimeException("不是學生對象");
        Student s = (Student) o;
        if (this.age > s.age) {
            return 1;
        }
        if (this.age == s.age) {
            return this.name.compareTo(s.name);
        }
        return -1;
    }
}

存入和取出順序一樣：comoareTo  return  正數

存入和取出順序相反：comoareTo  return  負數

只存入一個元素：comoareTo  return  0

當元素自身不具備比較性，或者具備的比較性不是所需要的，這時需要讓容器自身具備比較性，定義比較器，將比較器對象作為參數傳遞給TreeSet集合的構造函數。

public class TreeSetDemo {
    public static void main(String[] args) {
        TreeSet ts = new TreeSet(new MyCompare());
        ts.add(new Student("lisi02", 2));
        ts.add(new Student("lisi01", 1));
        ts.add(new Student("lisi04", 4));
        ts.add(new Student("lisi03", 3));
        ts.add(new Student("lisi04", 3));
        Iterator it = ts.iterator();
        while (it.hasNext()) {
            Student s = (Student) it.next();
            System.out.println(s.getName() + "::" + s.getAge());
        }
    }
}

class 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 int getAge() {
        return age;
    }
}

class MyCompare implements Comparator {

    @Override
    public int compare(Object o1, Object o2) {
        Student s1 = (Student) o1;
        Student s2 = (Student) o2;
        int num = s1.getName().compareTo(s2.getName());
        if (num == 0)
            return s1.getAge() - s2.getAge();
        return num;
    }
}

當兩種方式都存在時以比較器為主。

比較器：定義一個類，實現Comparetor介面，覆蓋compare方法。

存儲字元串並按長度排序：

字元串本身具備比較性，但它的比較性不是我們所需要的。使用比較器。

public class TreeSetDemo {
    public static void main(String[] args) {
        TreeSet ts = new TreeSet(new MyCompare());
        ts.add("sd");
        ts.add("q");
        ts.add("qq");
        ts.add("hsaa");
        ts.add("ehhhhh");
        Iterator it = ts.iterator();
        while (it.hasNext()) {
            System.out.println(it.next());
        }
    }
}

class MyCompare implements Comparator {

    @Override
    public int compare(Object o1, Object o2) {
        String s1 = (String) o1;
        String s2 = (String) o2;
        int num = new Integer(s1.length()).compareTo(new Integer(s2.length()));
        if (num == 0) {
            return s1.compareTo(s2);
        }
        return num;
    }
}