TreeSet:可以对Set集合中的元素进行排序。底层数据结构是二叉树,保证元素唯一性的依据,compareTo方法return 0;
TreeSet排序的第一种方式:让元素自身具备比较性。元素需要实现Comparable接口,覆盖compareTo方法。这种方式也称为元素的自然顺序,或者叫做默认顺序。
TreeSet的第二种排序方式:让元素自身不具备比较性时。或者具备的比较性不是所需要的。这时就需要让集合自身具备比较性。在集合初始化时,就有了比较方式。
import java.util.Iterator;
import java.util.TreeSet;
public class TreeSetDemo {
public static void main(String[] args) {
TreeSet ts = new TreeSet();
ts.add("cba");
ts.add("aaa");
ts.add("bca");
ts.add("Dbca");
Iterator it = ts.iterator();
while (it.hasNext())
{
System.out.println(it.next());
}
}
}
//输出
Dbca
aaa
bca
cba
需求:往TreeSet集合中存储自定义对象学生。
想按照学生的年龄进行排序。
记住:排序时,当主要条件相同时,一定判断一下次要条件。
import java.util.Iterator;
import java.util.TreeSet;
class Student0
{
private String name;
private int age;
Student0(String name,int age)
{
this.name = name;
this.age = age;
}
public String getName()
{
return name;
}
public int getAge()
{
return age;
}
}
public class TreeSetDemo0 {
public static void main(String[] args) {
TreeSet ts = new TreeSet();
ts.add(new Student0("lisi02",22));
ts.add(new Student0("lisi007",20));
ts.add(new Student0("lisi09",19));
ts.add(new Student0("lisi01",40));
Iterator it = ts.iterator();
while (it.hasNext())
{
Student0 stu = (Student0)it.next();
System.out.println(stu.getName()+"..."+stu.getAge());
}
}
}
//输出
Exception in thread "main" java.lang.ClassCastException: Student0 cannot be cast to java.base/java.lang.Comparable
at java.base/java.util.TreeMap.compare(TreeMap.java:1291)
at java.base/java.util.TreeMap.put(TreeMap.java:536)
at java.base/java.util.TreeSet.add(TreeSet.java:255)
at TreeSetDemo0.main(TreeSetDemo0.java:25)
import java.util.Iterator;
import java.util.TreeSet;
class Student0 implements Comparable//该接口强制让学生具备比较性。
{
private String name;
private int age;
Student0(String name,int age)
{
this.name = name;
this.age = age;
}
public String getName()
{
return name;
}
public int getAge()
{
return age;
}
public int compareTo(Object obj)
{
if (!(obj instanceof Student0))
throw new RuntimeException("不是学生对象");
Student0 s = (Student0)obj;
System.out.println(this.name + "...compareto....."+s.name);
if (this.age > s.age)
return 1;
if (this.age == s.age)
return (this.name.compareTo(s.name));
return -1;
}
}
public class TreeSetDemo0 {
public static void main(String[] args) {
TreeSet ts = new TreeSet();
ts.add(new Student0("lisi02",22));
ts.add(new Student0("lisi007",20));
ts.add(new Student0("lisi09",19));
ts.add(new Student0("lisi01",40));
ts.add(new Student0("lisi06",19));
ts.add(new Student0("lisi01",40));
Iterator it = ts.iterator();
while (it.hasNext())
{
Student0 stu = (Student0)it.next();
System.out.println(stu.getName()+"..."+stu.getAge());
}
}
}
//输出
lisi02...compareto.....lisi02
lisi007...compareto.....lisi02
lisi09...compareto.....lisi02
lisi09...compareto.....lisi007
lisi01...compareto.....lisi007
lisi01...compareto.....lisi02
lisi06...compareto.....lisi007
lisi06...compareto.....lisi09
lisi01...compareto.....lisi007
lisi01...compareto.....lisi02
lisi01...compareto.....lisi01
lisi06...19
lisi09...19
lisi007...20
lisi02...22
lisi01...40
当元素自身不具备比较性,或者具备的比较性不是所需要的。这时需要让容器自身具备比较性。定义了比较器,将比较器对象作为参数传递给TreeSet集合的构造函数。
当两种排序都存在时,以比较器为主。定义一个类,实现Compatator接口,覆盖compare方法。
import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;
class Student1
{
private String name;
private int age;
Student1(String name,int age)
{
this.name = name;
this.age = age;
}
public String getName()
{
return name;
}
public int getAge()
{
return age;
}
}
public class TreeSetDemo1 {
public static void main(String[] args) {
TreeSet ts = new TreeSet(new MyCompare());
ts.add(new Student1("lisi02",22));
ts.add(new Student1("lisi007",20));
ts.add(new Student1("lisi09",19));
ts.add(new Student1("lisi01",40));
ts.add(new Student1("lisi06",19));
ts.add(new Student1("lisi01",40));
Iterator it = ts.iterator();
while (it.hasNext())
{
Student1 stu = (Student1)it.next();
System.out.println(stu.getName()+"..."+stu.getAge());
}
}
}
class MyCompare implements Comparator
{
public int compare(Object o1,Object o2)
{
Student1 s1 = (Student1) o1;
Student1 s2 = (Student1) o2;
int num = s1.getName().compareTo(s2.getName());
if (num == 0)
{
if (s1.getAge()> s2.getAge())
return 1;
if (s1.getAge() == s2.getAge())
return 0;
return -1;
}
return num;
}
}
//输出
lisi007...20
lisi01...40
lisi02...22
lisi06...19
lisi09...19
调用Integer对象的compareTo方法进行比较
import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;
class Student2
{
private String name;
private int age;
Student2(String name,int age)
{
this.name = name;
this.age = age;
}
public String getName()
{
return name;
}
public int getAge()
{
return age;
}
}
public class TreeSetDemo2 {
public static void main(String[] args) {
TreeSet ts = new TreeSet(new MyCompare0());
ts.add(new Student2("lisi02",22));
ts.add(new Student2("lisi007",20));
ts.add(new Student2("lisi09",19));
ts.add(new Student2("lisi01",40));
ts.add(new Student2("lisi06",19));
ts.add(new Student2("lisi01",40));
ts.add(new Student2("lisi01",39));
Iterator it = ts.iterator();
while (it.hasNext())
{
Student2 stu = (Student2)it.next();
System.out.println(stu.getName()+"..."+stu.getAge());
}
}
}
class MyCompare0 implements Comparator
{
public int compare(Object o1,Object o2)
{
Student2 s1 = (Student2) o1;
Student2 s2 = (Student2) o2;
int num = s1.getName().compareTo(s2.getName());
if (num == 0)
{
return new Integer(s1.getAge()).compareTo(new Integer(s2.getAge()));
}
return num;
}
}
//输出
lisi007...20
lisi01...39
lisi01...40
lisi02...22
lisi06...19
lisi09...19
按照字符串长度排序
字符串本身具备比较性,但是它的比较方式不是所需要的。
这时就只能使用比较器。
import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;
public class TreeSetDemo3 {
public static void main(String[] args) {
TreeSet ts = new TreeSet(new MyCompare1());
ts.add("abcd");
ts.add("cc");
ts.add("cba");
ts.add("aaa");
ts.add("z");
ts.add("hahaha");
Iterator it = ts.iterator();
while (it.hasNext())
{
System.out.println(it.next());
}
}
}
class MyCompare1 implements Comparator
{
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;
}
}
//输出
z
cc
aaa
cba
abcd
hahaha
匿名内部类写法
import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;
public class TreeSetDemo4 {
public static void main(String[] args) {
TreeSet ts = new TreeSet(new 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;
}
});
ts.add("abcd");
ts.add("cc");
ts.add("cba");
ts.add("aaa");
ts.add("z");
ts.add("hahaha");
Iterator it = ts.iterator();
while (it.hasNext())
{
System.out.println(it.next());
}
}
}
//输出
z
cc
aaa
cba
abcd
hahaha
倒序排列
import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;
public class TreeSetDemo5 {
public static void main(String[] args) {
TreeSet ts = new TreeSet(new MyCompare2());
ts.add("abcd");
ts.add("cc");
ts.add("cba");
ts.add("aaa");
ts.add("z");
ts.add("hahaha");
Iterator it = ts.iterator();
while (it.hasNext())
{
System.out.println(it.next());
}
}
}
class MyCompare2 implements Comparator
{
public int compare(Object o1,Object o2)
{
String s1 = (String)o1;
String s2 = (String)o2;
System.out.println(s1+"...compareTo..."+s2);
int num = new Integer(s2.length()).compareTo(new Integer(s1.length()));
if (num == 0)
{
return s2.compareTo(s1);
}
return num;
}
}
//输出
abcd...compareTo...abcd
cc...compareTo...abcd
cba...compareTo...abcd
cba...compareTo...cc
aaa...compareTo...cba
aaa...compareTo...cc
z...compareTo...cba
z...compareTo...cc
hahaha...compareTo...cba
hahaha...compareTo...abcd
hahaha
abcd
cba
aaa
cc
z
网友评论