TreeSet 是sortedSet的唯一实现类,正如SortedSet名字暗示,TreeSet可以让集合元素处在排好序的状态。
与HashSet相比,TreeSet还额外提供了以下的方法(列出来,混个脸熟:)
SortedSet subSet(Object fromElement,Object toElement) :返回这个Set的子集合,范围从fromElement(包含)到toElement(不包含)
SortedSet headSet(Object toElement):返回这个Set的子集合,范围小于到toElement的子集合
SortedSet tailSet(Object fromElement):返回这个Set的子集合,范围大于或等于到fromElement的子集合
Object first(): 返回这个Set第一个元素
Object last():返回这个Set最后一个元素
Object lower(Object e):返回小于指定元素的集合里最大的那个元素
Object higher(Object e):返回大于指定元素的集合里最小的那个元素
【以上参考元素都不需要是集合里的】
总结一下,最后四个方法就是找到集合里的第一个,前一个,后一个,最后一个元素。同时前三个就是返回该集合的符合条件的子集合。
package Test01;
import java.util.TreeSet;
public class TestTreeSet {
public static void main(String[] args) {
TreeSet num =new TreeSet();
num.add(2);
num.add(-2);
num.add(10);
num.add(9);
System.out.println(num); //看出:不是按照添加的顺序来,是数字由小到大排序
System.out.println(num.first()); //看出:数字由小到大排序的第一个
System.out.println("9到12 之间的"+num.subSet(9, 12));
System.out.println("比8小的"+num.headSet(8));
System.out.println("比8大的"+num.tailSet(8));
}
}
与hashset采用hash算法决定元素的存储位置,TreeSet采用红黑树的数据结构(待跟进)来存储集合元素。那么他的排序规则是怎么的呢?
自然排序(默认情况)。
调用TreeSet的无参的构造器,并实现Comparable接口,实现里的 compareTo(Object obj)方法,来比较集合元素的大小,然后按照compareTo里的排序。
该种方法注意:
向TreeSet里添加元素时,只有第一个无须实现Comparable接口,后面的必须实现。(第二个及后面的会调用compareTo方法,与集合里的其他元素比较,要求了比较的两个元素都是一个类的实例,否则会ClassCastException)
package Test01;
import java.util.TreeSet;
public class Foo implements Comparable{
int num;
public Foo(int num) {
this.num =num;
}
public int getNum() {
return num;
}
public void setNum(int num) {
this.num = num;
}
@Override
public String toString() {
// TODO Auto-generated method stub
return "foo:"+this.getNum();
}
@Override
public int compareTo(Object obj) {
// TODO Auto-generated method stub
if(obj instanceof Foo) {
Foo f = (Foo) obj;
if(this.num > f.getNum()) {
return 1;
}
if(this.num == f.getNum()) {
return 0;
}
else
return -1;
}
else return 0;
}
public static void main(String[] args) {
TreeSet<Foo> test = new TreeSet<>();
test.add(new Foo(8)); test.add(new Foo(3));
System.out.println(test);
}
}
下面的一些常用的类已经实现了Comparable接口,提供了比较大小的标准:
BigDecimal
character
Boolean
String
Date Time
对于TreeSet集合来说,判断两个对象是否相等的唯一标准 是compareTo(Object obj)返回0,是0 就相等,不是0就不等。就不会让相等的对象进入集合。
举例
(正确的结果)能够插入两个compare 不等于0但是equals为true的对象
package Test01;
import java.util.Comparator;
import java.util.TreeSet;
public class Comp implements Comparator<Foo>{
@Override
public int compare(Foo o1, Foo o2) {
return 1;
}
@Override
public boolean equals(Object obj) {
// TODO Auto-generated method stub
return true;
}
public static void main(String[] args) {
TreeSet<Foo> t = new TreeSet<>(new Comp());
Foo f = new Foo(0);
t.add(f);
t.add(f);
System.out.println(t);
t.first().num =9;
System.out.println(t);
}
}
留意下注释的差别,(问题已解决)
package Test01;
import java.util.TreeSet;
public class Test implements Comparable{
int num;
public Test(int num) {
this.num=num;
}
public int getNum() {
return num;
}
public void setNum(int num) {
this.num = num;
}
@Override
public boolean equals(Object obj) {
// TODO Auto-generated method stub
return true;
}
@Override
public int compareTo(Object o) {
// TODO Auto-generated method stub
return 1;
}
public static void main(String[] args) {
TreeSet t = new TreeSet();
Test f = new Test(0);
t.add(f);
System.out.println("会把同一个对象但是compareTo返回1添加进TreeSet吗?"+ t.add(f)); //equals不是评价对象相等的标准,compareTo返回0才相等。返回true
System.out.println(t);
((Test)(t.first())).num=9; //修改第一个元素,下面一行发现最后一个元素也被改成了9
System.out.println( ((Test)(t.last())).num);
}
}
由正确的结果注意:TreeSet也是Set,equals结果注意保持和compareTo结果一致。因为如果compareTo 等于0,但是equals发现false,就和Set冲突。
和hashset同样,讨论下,可变对象插入删除问题(留意注释部分)
package Test01;
import java.util.Iterator;
import java.util.TreeSet;
class mutClass implements Comparable{
public int count;
public mutClass(int count) {
this.count =count;
}
public boolean equals(Object obj) {
if(this == obj) {
return true;
}
if(obj != null && obj.getClass() == mutClass.class) {
mutClass m =(mutClass) obj;
return this.count == m.count;
}
return false;
}
public int hashcode() {
return this.count;
}
public String toString() {
return "试试mutClass[count=" + count + "]";
}
@Override
public int compareTo(Object o) {
// TODO Auto-generated method stub
mutClass m = (mutClass) o;
return count>m.count?1 :count<m.count?-1 :0;
}
}
public class TestTreeSet {
@SuppressWarnings("unchecked")
public static void main(String[] args){
TreeSet testHashSet =new TreeSet();
mutClass a = new mutClass(3);
mutClass b = new mutClass(1);
mutClass c = new mutClass(-9);
mutClass d = new mutClass(9);
testHashSet.add(a);
testHashSet.add(b);
testHashSet.add(c);
testHashSet.add(d);
System.out.println("第一次"+testHashSet);
((mutClass)testHashSet.first()).count =20;
System.out.println("第二次"+testHashSet); //修改可变对象的实例变量,不会调整顺序
((mutClass)testHashSet.last()).count =1; //修改可变对象的实例变量,不会调整顺序,且可能出现重复对象的情况
System.out.println("第三次"+testHashSet);
System.out.println(testHashSet.remove(new mutClass(1))); //删除实例变量被修改的元素,应该删除失败,但是我删除成功了?
System.out.println("第四次"+testHashSet);
}
}
定制排序。
自己写个比较器的类,传入TreeSet的有参数的构造器,这样往集合里add元素就能按照定制的顺序排序了。
package Test01;
import java.util.Comparator;
import java.util.TreeSet;
public class Comp implements Comparator<Foo>{
@Override
public int compare(Foo o1, Foo o2) {
// TODO Auto-generated method stub
if(o1.getNum()>o2.getNum()) {
return 1;
}
if (o1.getNum() == o2.getNum()) {
return 0;
}
else return -1;
}
public static void main(String[] args) {
TreeSet<Foo> t = new TreeSet<>(new Comp());
t.add(new Foo(0));
t.add(new Foo(-9));
System.out.println(t);
}
}