集合TreeSet的使用
集合中的TreeSet是集合体系结构中的底层实现,是Collection的孙子,Set的儿子。TreeSet除拥有父接口的特点外,还有其自身的特点。下面就看看TreeSet的排序是怎么实现的。从它的构造方法看,提供了无参和带参两种。
常用到的两个构造:
无参构造 public TreeSet()
构造一个新的空set,该set根据其元素的自然顺序进行排序。插入该set的所有元素都必须实现Comparable接口。
带参构造 public TreeSet(Comparator<? super E> comparator)
构造一个新的空TreeSet,它根据指定比较器进行排序。
API文档里面已经描述的很清楚,要实现排序,要么实现Comparable接口,要么指定比较器Comparator。接下来就分别用两种方式来实现这个过程。
描述一个Person类。拥有成员:name、age、有参、无参构造、getters、setters。
package cn.dolphin; public class Person { private String name; private int age; public Person() {super();} public Person(String name, int age) { super(); 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; } }
第一种方式:
使用有参构造,传递Comparator比较器。而Comparator是接口,所以没有对象,这时传递的就是实现了Comparator接口的子类的对象,而如果这个对象只用到一次又可以写成匿名内部类,于是就有了下面的代码。
package cn.dolphin; import java.util.Comparator; import java.util.TreeSet; public class TreeSetDemo { public static void main(String[] args) { // 创建TreeSet对象,用来存储Person类型对象。 TreeSet<Person> ts = new TreeSet<>(new Comparator<Person>() { @Override public int compare(Person o1, Person o2) { int num1 = o1.getName().length() - o2.getName().length(); int num2 = num1 == 0 ? o1.getName().compareTo(o2.getName()) : num1; int num3 = num2 == 0 ? o1.getAge() - o2.getAge() : num2; return num3; } }); // 创建Person对象并添加到ts集合。 ts.add(new Person("aidengbao", 30)); ts.add(new Person("bosideng", 21)); ts.add(new Person("yichun", 18)); ts.add(new Person("yichun", 18)); for (Person s : ts) { System.out.println(s.getName() + ":" + s.getAge()); } } }
当然也可不写成匿名内部类,只是这样就必须得定义一个类来实现Comparator接口,于是用下面的代码:
package cn.dolphin; import java.util.Comparator; public class ComparatorImplements implements Comparator<Person>{ @Override public int compare(Person o1, Person o2) { //主要条件,按照Person的姓名长度排序,如不相同,则添加到集合。 int num1 = o1.getName().length() - o2.getName().length(); //如果姓名长度一样,再比较姓名是否相同。如不相同,则添加到集合。 int num2 = num1 == 0 ? o1.getName().compareTo(o2.getName()) : num1; //如果姓名也相同,再比较年龄是不是相同,如果相同,则不添加到集合,保证数据的唯一。 int num3 = num2 == 0 ? o1.getAge() - o2.getAge() : num2; return num3; } }
这样,有了ComparatorImplements类实现Comparator接口后,就可以用多态的方式创建ComparatorImplements的对象传递给TreeSet。接着看代码:
package cn.dolphin; import java.util.Comparator; import java.util.TreeSet; public class TreeSetDemo { public static void main(String[] args) { // 创建TreeSet对象,用来存储Person类型对象。 Comparator<Person> ci = new ComparatorImplements(); TreeSet<Person> ts = new TreeSet<>(ci); // 创建Person对象并添加到ts集合。 ts.add(new Person("aidengbao", 30)); ts.add(new Person("bosideng", 21)); ts.add(new Person("yichun", 18)); ts.add(new Person("yichun", 18)); for (Person s : ts) { System.out.println(s.getName() + ":" + s.getAge()); } } }
通过上面的代码,明显看到使用匿名内部类的好处就是简单方便,接下来,再看看用无参构造的实现。
第二种方式:
使用无参构造,Person类自己实现Comparable接口。
package cn.dolphin; public class Person implements Comparable<Person>{ private String name; private int age; public Person() {super();} public Person(String name, int age) { super(); 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; } @Override public int compareTo(Person o) { int num1 = this.name.length() - o.name.length(); int num2 = num1 == 0 ? this.name.compareTo(o.name) : num1; int num3 = num2 == 0 ? this.age - o.age : num2; return num3; } }
好像看起来这样要好一点点,因为在这里可以直接使用name和age成员变量,而上面只能过过getName()和getAge()来获取,因为定义的成员在开发中可能都是私有的。Person类实现了Comparable接口,就可以使用无参的构造来对对象进行唯一性判断和对集合排序。
package cn.dolphin; import java.util.TreeSet; public class TreeSetDemo { public static void main(String[] args) { // 创建TreeSet对象,用来存储Person类型对象。 TreeSet<Person> ts = new TreeSet<>(); // 创建Person对象并添加到ts集合。 ts.add(new Person("aidengbao", 30)); ts.add(new Person("bosideng", 21)); ts.add(new Person("yichun", 18)); ts.add(new Person("yichun", 18)); for (Person s : ts) { System.out.println(s.getName() + ":" + s.getAge()); } } }
虽然写了这么长,但是TreeSet并不是多么重要,用到并不多,所以也就当个了解内容吧,不过多知道点儿总归不是坏事儿。