黑马程序员——集合框架（TreeSet）

-----------android培训、java培训、java学习型技术博客、期待与您交流！------------

Set：无序，不可以重复元素。
|——HashSet：数据结构是哈希表。线程是非同步的。
保证元素唯一性的原理：判断元素的hashCode是否相同。
如果相同，还会继续判断元素的equals方法，是否为true。
|——TreeSet：可以对Set集合中的元素进行排序。
底层数据结构是二叉树。
保证元素唯一性的依据。
通过对象中的compareTo方法确定元素的大小，和确定是否相同。

TreeSet排序的第一种方式：让元素自身具备比较性。元素需要实现Comparable接
口，覆盖compareTo方法。这种方式也称为元素的自然顺序，或者叫作默认顺
序。

TreeSet集合的第二种排序方式。
当元素自身不具备比较性时，或者具备的比较性不是所需要的。
这时就需要让集合自身具备比较性。
在集合初始化时就有了比较方式。

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("abcd");

Iterator it = ts.iterator();
while(it.hasNext())
System.out.println(it.next());
}
}

控制台显示结果：
aaa
abcd

bca

cba

有结果可知TreeSet默认实现自然排序。

向TreeSet集合中添入自定义对象。
此对象必须实现Comparable接口的方法，compareTo（）；
排序时，当主要条件相同时，一定要判断一下次要条件。
程序举例：

import java.util.Iterator;
import java.util.TreeSet;

public class TreeSetDemo {

public static void main(String[] args)
{

TreeSet<Student> ts = new TreeSet<Student>();

ts.add(new Student("lisi01",18));
ts.add(new Student("lisi02",19));
ts.add(new Student("lisi04",18));
ts.add(new Student("lisi04",18));

Iterator<Student> it = ts.iterator();
while(it.hasNext())
{
Student stu = it.next();
System.out.println(stu.getName()+"......"+stu.getAge());
}

}
}

class Student  implements Comparable<Student>//该接口强制让学生具备比较性。
{
private String name;
private int age;
Student(String name,int age)
{
this.setName(name);
this.setAge(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(Student o) {
if(this.age>o.age)
return 1;
else if(this.age==o.age)
return this.name.compareTo(o.getName());
else
return -1;
}

}

TreeSet底层数据结构：
红黑树，二叉树。自动取则中值。
取值：
从最左边开始取。每个一个节点的左边必小于右边。



如果将自定义对象的compareTo方法复写成：
public int compareTo（Object obj）
｛
return 1；
｝
则TreeSet里面存自定义对象时，都认为存入的下一个值比上一个值大，都存在二叉树的右边。
在获取该TreeSet里面的数据时，会将存入的先后值依次打印出，也就是怎么存入怎么取出。

import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;

/*
* 当元素自身不具备比较性，或者具备的比较性不是所需要的。
* 这时需要让容器自身具备比较性。
* 定义了比较器，将比较器对象作为参数传递给TreeSet集合的构造函数。
*
* 当两种排序都存在时，以比较器为主。
*
* 定义一个类，实现Comparator接口，覆盖compare方法。
*
*
*/
public class TreeSetDemo2 {

public static void main(String[] args) {

Comparator<Student> com= new Comparator<Student>(){
//这个类本身就继承了equals方法，所以可以不覆盖equals方法。
@Override
public int compare(Student s1, Student s2) {
// TODO 自动生成的方法存根
if(s1.getName().compareTo(s2.getName())!=0)
{
return s1.getName().compareTo(s2.getName());
}
else if(s1.getAge()>s2.getAge())
{
return 1;
}
else if(s1.getAge()<s2.getAge())
{
return -1;
}
else
{
return 0;
}
}
//return new Integer(s1.getAge()).compareTo(new Integer(s2.getAge());
};

TreeSet<Student> ts = new TreeSet<Student>(com);

ts.add(new Student("lisi01",18));
ts.add(new Student("lisi02",19));
ts.add(new Student("lisi04",18));
ts.add(new Student("lisi001",19));

Iterator<Student> it = ts.iterator();
while(it.hasNext())
{
Student stu = it.next();
System.out.println(stu.getName()+"......"+stu.getAge());
}

}

}

class Student  implements Comparable<Student>//该接口强制让学生具备比较性。
{
private String name;
private int age;
Student(String name,int age)
{
this.setName(name);
this.setAge(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(Student o) {
if(this.age>o.age)
return 1;
else if(this.age==o.age)
return this.name.compareTo(o.getName());
else
return -1;
}

}

练习： 按照字符串长度排序

import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;

public class TreeSetTest {

public static void main(String[] args) {

//自定义比较方式。
Comparator<String> mycomp = new Comparator<String>(){

@Override
public int compare(String s1, String s2) {
int num = new Integer(s1.length()).compareTo(s2.length());
if(num == 0)
return s1.compareTo(s2);
return num;
}
};
TreeSet<String> ts = new TreeSet<String>(mycomp);

ts.add("abcd");
ts.add("ccc");
ts.add("cba");
ts.add("z");
ts.add("hahaha");

Iterator<String> it = ts.iterator();
while(it.hasNext())
{
System.out.println(it.next());
}

}

}