黑马程序员-java之TreeSet,Comparable,Comparator

-------------android培训、java培训、---------------

TreeSet可以对元素进行排序，缺省是按照自然排序进行排列，即元素必须实现Comparable接口，或者创建TreeSet对象的时候必须指定一个比较器

且元素唯一。

排序的两种方式：

方式一：让元素自身具备比较性，即让元素类实现Comparable接口，覆写compareTo(),这种排序方式是默认排序，也叫自然排序

方式二：当元素自身不具备比较性时，或者所具备的比较性不是实际需要的，这是就需要让集合TreeSet具备比较性，即在创建集合对象的时候指定比较器

当两种都存在时候，用比较器

原理:二叉树结构（左叉值小于右叉值）,新元素大于原有元素，则放在叉的右边，小于则放在左边，等于则不存储

实例：存储自定义对象

代码：

package com.itheima;

import java.util.Comparator;

import java.util.Iterator;

import java.util.TreeSet;

class Student implements Comparable{

private String name;

private int age;

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(Object obj){//先按年龄比较，年龄相同则按名字自然排序

if(!(obj instanceof Student)){

throw new RuntimeException(obj.toString()+"不是学生");

}

Student stu = (Student)obj;

//System.out.println(this.name+"---compareTo---"+stu.name);

if(this.age>stu.age)

return 1;

if(this.age==stu.age){

return this.name.compareTo(stu.name);

}

return -1;

}

}

class myComparator implements Comparator{//先按姓名自然排序，姓名相同则按年龄升序排序

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 new Integer(s1.getAge()).compareTo(s2.getAge());

}

return num;

}

}

public class TreeSetDemo {

public static void main(String[] args) {

// TODO Auto-generated method stub

TreeSet ts = new TreeSet();//创建TreeSet对象

TreeSet ts2 = new TreeSet(new myComparator());

ts.add(new Student("zhangsan",18));//添加元素到对象ts

ts.add(new Student("lisi",22));

ts.add(new Student("wangwu",18));

ts.add(new Student("zhaoliu",19));

ts2.add(new Student("zhangsan",18));//添加元素到ts2

ts2.add(new Student("lisi",22));

ts2.add(new Student("wangwu",18));

ts2.add(new Student("zhaoliu",19));

ts2.add(new Student("wangwu",23));

Iterator it = ts.iterator();

Iterator it2 = ts2.iterator();

System.out.println("----------ts中的元素------------");

while(it.hasNext()){

Student stu = (Student)it.next();

System.out.println(stu.getName()+"---"+stu.getAge());

}

System.out.println("----------ts2中的元素------------");

while(it2.hasNext()){

Student stu2 = (Student)it2.next();

System.out.println(stu2.getName()+"---"+stu2.getAge());

}

}

}