不用多说,我先贴上一个示例代码,然后再解释:
package com.ccc.java2.collections.app;
import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;
public class Student implements Comparable<Student> {
private int id;
private String name;
public Student(int id, String name) {
this.id = id;
this.name = name;
}
public void setId(int id) {
this.id = id;
}
public void setName(String name) {
this.name = name;
}
public int getId() {
return id;
}
public String getName() {
return name;
}
/*
* Student 类的字符串表达式,形如: 2 张三
*/
public String toString() {
return (id + "\t" + name);
}
/*
* 实现 Comparable 接口中的 compareTo 方法, 通常大于时返回一个正数,小于时返回一个负数,
* 等于时返回零,具体情况可以自行决定。
* ******************************************************** 这里我根据 id
* 号的大小进行了比较。由于 TreeSet 会根据 compareTo 的结果来排序,因此输出结果 应该是按照 id 号从小到大排序的。
* 如果要根据姓名进行排序,只需对这个方法进行相应的修改。
*/
/***
* public int compareTo(T o); 超類使用泛型
*/
public int compareTo(Student arg) {
if (id > arg.id)
return 1;
else if (id == arg.id)
return 0;
else
return -1;
}
//兩種不同的排序方式
public static void main(String args[]) {
TreeSet<Student> sortByDescription = new TreeSet<Student>(
new Comparator<Student>() {
@Override
public int compare(Student o1, Student o2) {
// TODO Auto-generated method stub
String name1 = o1.getName();
String name2 = o2.getName();
return name1.compareTo(name2);
}
});
sortByDescription.add(new Student(5, "Tom"));
sortByDescription.add(new Student(3, "John"));
sortByDescription.add(new Student(9, "David"));
sortByDescription.add(new Student(7, "Alice"));
Iterator<Student> iteratorByNameIterator = sortByDescription.iterator();
while (iteratorByNameIterator.hasNext()) {
System.out.println(iteratorByNameIterator.next().toString());
}
System.out.println("另外一種排序方式...........");
// System.out.println();
// System.err.println("err");
TreeSet<Student> tset = new TreeSet<Student>();
tset.add(new Student(5, "Tom"));
tset.add(new Student(3, "John"));
tset.add(new Student(9, "David"));
tset.add(new Student(7, "Alice"));
Iterator<Student> itor = tset.iterator();
while (itor.hasNext()) {
System.out.println(itor.next().toString());
}
}
}
里面的代码比较多,我讲里面一些代码段:
TreeSet<Student> sortByDescription = new TreeSet<Student>(
new Comparator<Student>() {
@Override
public int compare(Student o1, Student o2) {
// TODO Auto-generated method stub
String name1 = o1.getName();
String name2 = o2.getName();
return name1.compareTo(name2);
}
});
这段代码,实现的是Comparator类的compare方法。