当很多人问我读研到底好不好的时候,我总是说上研很苦逼,读完研之后都不知道自己能不能找到工作,所以不建议同学们读研~即使要读也读一个985或者211的研究生,这是我肺腑之言。但还有一半我没说完,读研的时候你可能会找到你喜欢的活动,会遇到一些愿意和你一起玩的玩伴,在读研期间可以很任性的在想玩耍的时候就去玩儿,这是读研的福利,很高兴我能遇到。好了开始技术!
学习Java,肯定会学到集合,然后遇到TreeSet,我现在就遇到了它。同时使用它的时候遇到一些问题。首先说说什么是TreeSet。TreeSet是Set 的子接口,而Set接口的元素不可重复、无序的特性TreeSet也是具备的。另外,TreeSet还具有可以对集合中的元素进行指定顺序的排序的功能。
这个功能实现的基本原理就是,当存放元素的时候进行比较,比较之后再存入。对于普通的字符串它可以完美的实现排序功能:
public static void demo1() {
TreeSet ts = new TreeSet();
ts.add("abc");
ts.add("nba");
ts.add("linweieran");
ts.add("cba");
Iterator it = ts.iterator();
while(it.hasNext()){
System.out.println(it.next());
}
}
执行结果如下:
运行结果正确。
但当我向TreeSet集合中保存自定义的Person对象是就会发生错误!
public static void main(String[] args) {
TreeSet ts = new TreeSet(new CompareByName()) ;
ts.add(new Person("linweieran",24));
ts.add(new Person("linweieran1",23)); //bean.Person cannot be cast to java.lang.Comparable
ts.add(new Person("linweieran2",22));
ts.add(new Person("weixiaobao",25));
ts.add(new Person("linweieran",24));
Iterator it = ts.iterator();
while(it.hasNext()){
Person p = (Person)it.next();
System.out.println(p.getName()+":"+p.getAge());
}
}
发生这样的错误:查文档会找到这样的解释:Comparable此接口强行对实现它的每个类的对象进行强行排序,这种排序成为自然排序,类的compareTo 方法就是它的自然比较方法。所以为了实现该功能就有两个方法
TreeSet对元素进行排序的方式一:让元素自身具备比较功能,元素需要实现Comparable接口,覆盖compareTo()方法;简而言之就是在Person类中继承Comparable接口,并重写compareTo()方法。
public class Person implements Comparable {
private String name;
private int age;
//alt+shift+s
public Person(String name, int age) {
super();
this.name = name;
this.age = age;
}
/*
* 以Person对象的年龄进行从小到大的排序。
*/
@Override
public int compareTo(Object o) {
//进行比较,
Person p = (Person)o;
/*
if(this.age>p.age)
return 1;
if(this.age<p.age)
return -1;
if(this.age==p.age){
//姓名怎么比较-----
return this.name.compareTo(p.name);
}
*/
System.out.println(this.name+"---compareTo---");
int temp= this.age-p.age;
return temp==0?this.name.compareTo(p.name):temp;
}
编码思路是:先对年龄进行比较,如果年龄相容再对姓名(String)进行比较,最后可以简约成三行代码!!
这样子就可以实现排序功能。
但是,使用上述的方式有一个问题,如果Person类不是自己写的,我们就无法在Person类中进行修改实现该功能,如果是这种情况怎么办呢?这就需要另一种方式了!
TreeSet集合的第二种排序方式:让集合自身具备比较功能,定义一个类实现Comparator接口,覆盖Comparat方法。构造函数TreeSet(Comparator c)的比较器。
首先我们需要写一个类,该类继承Comparator,同时实现按姓名排序的方法。
public class CompareByName implements Comparator {
@Override
public int compare(Object o1, Object o2) {
System.out.println("CompareByName----");
Person p1=(Person)o1;
Person p2=(Person)o2;
int temp = p1.getName().compareTo(p2.getName());
return temp==0?p1.getAge()-p2.getAge():temp;
}
}
然后,在创建TreeSet的时候需要直接用到TreeSet(Comparator c)的构造函数。
public static void main(String[] args) {
/*
* 以Person对象的年龄进行从小到大的排序。
*/
TreeSet ts = new TreeSet(new CompareByName()) ;//new CompareByName()
ts.add(new Person("linweieran",24));
ts.add(new Person("linweieran1",23)); //bean.Person cannot be cast to java.lang.Comparable
ts.add(new Person("linweieran2",22));
ts.add(new Person("weixiaobao",25));
ts.add(new Person("linweieran",24));
Iterator it = ts.iterator();
while(it.hasNext()){
Person p = (Person)it.next();
System.out.println(p.getName()+":"+p.getAge());
}
}
这样子就可以实现排序功能了。
哈哈,运行结果正确!!!