java treeset_java treeset

1.TreeSet原理:

/*

* TreeSet存储对象的时候, 可以排序, 但是需要指定排序的算法

*

* Integer能排序(有默认顺序), String能排序(有默认顺序), 自定义的类存储的时候出现异常(没有顺序)

*

* 如果想把自定义类的对象存入TreeSet进行排序, 那么必须实现Comparable接口

* 在类上implement Comparable

* 重写compareTo()方法

* 在方法内定义比较算法, 根据大小关系, 返回正数负数或零

* 在使用TreeSet存储对象的时候, add()方法内部就会自动调用compareTo()方法进行比较, 根据比较结果使用二叉树形式进行存储

*/

package test.treeset;

import java.util.Iterator;

import java.util.Set;

import java.util.TreeSet;

public class test_treeset {

@SuppressWarnings("unchecked")

public static void main(String[] args) {

Set ts = new TreeSet();

ts.add(new Teacher("zhangsan", 1));

ts.add(new Teacher("lisi", 2));

ts.add(new Teacher("wangmazi", 3));

ts.add(new Teacher("wangwu",4));

ts.add(new Teacher("mazi", 3));

Iterator it = ts.iterator();

while (it.hasNext()) {

System.out.println(it.next());

}

}

}

class Teacher implements Comparable {

int num;

String name;

Teacher(String name, int num) {

this.num = num;

this.name = name;

}

public String toString() {

return "学号：" + num + "\t\t姓名：" + name;

}

//o中存放时的红黑二叉树中的节点，从根节点开始比较

public int compareTo(Object o) {

Teacher ss = (Teacher) o;

int result = num < ss.num ? 1 : (num == ss.num ? 0 : -1);//降序

//int result = num > ss.num ? 1 : (num == ss.num ? 0 : -1);//升序

if (result == 0) {

result = name.compareTo(ss.name);

}

return result;

}

}

运行结果：

学号：4 姓名：wangwu

学号：3姓名：mazi

学号：3姓名：wangmazi

学号：2姓名：lisi

学号：1姓名：zhangsan

3.比较器

在使用Arrays对数组中的元素进行排序的时候，可以传递一个比较器。

在使用Collections对集合中的元素进行排序的时候，可以传递一个比较器。

那么在使用TreeSet对加入到其中的元素进行排序的时候可以传入一个比较器吗？

public TreeSet(Comparator super E> comparator) {

this(new TreeMap(comparator));

}

通过查看它的构造方法就知道可以传入一个比较器。

构造一个新的空TreeSet，它根据指定比较器进行排序。插入到该 set 的所有元素都必须能够由指定比较器进行相互比较：对于 set 中的任意两个元素 e1 和e2，执行 comparator.compare(e1, e2) 都不得抛出 ClassCastException。如果用户试图将违反此约束的元素添加到 set 中，则 add 调用将抛出 ClassCastException。

package test.treeset;

import java.util.Comparator;

import java.util.Iterator;

import java.util.TreeSet;

public class TreeSetTest {

@SuppressWarnings("unchecked")

public static void main(String[] args) {

TreeSet ts = new TreeSet(new Teacher2.TeacherCompare());

ts.add(new Teacher2("zhangsan", 2));

ts.add(new Teacher2("lisi", 1));

ts.add(new Teacher2("wangmazi", 3));

ts.add(new Teacher2("mazi", 3));

Iterator it = ts.iterator();

while (it.hasNext()) {

System.out.println(it.next());

}

}

}

class Teacher2 {

int num;

String name;

Teacher2(String name, int num) {

this.num = num;

this.name = name;

}

public String toString() {

return "学号：" + num + " 姓名：" + name;

}

static class TeacherCompare implements Comparator {// 老师自带的一个比较器

//o1中存放的事目标节点

//o2中存放时的红黑二叉树中的节点，从根节点开始比较

public int compare(Object o1, Object o2) {

Teacher2 s1 = (Teacher2) o1;// 转型

Teacher2 s2 = (Teacher2) o2;// 转型

int result = s1.num > s2.num ? 1 : (s1.num == s2.num ? 0 : -1);

if (result == 0) {

result = s1.name.compareTo(s2.name);

}

return result;

}

}

}