treeSet剖析

2024-05-06 08:32

文章标签 剖析 treeset

本文主要是介绍treeSet剖析，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

TreeSet使用元素的自然顺序（暂时理解为unicode顺序）对集合中的元素进行排序。添加的元素必须自己实现Comparable接口，以便排序时默认（源码理解）调用其compareTo方法进行比较，TreeSet本身不须实现Comparable接口。

为保证元素唯一性，当有相同元素时，不能往里添加。判断元素唯一性的方式：根据比较方法返回的结果是不是0（public int compareTo(Object o)）。

//由于String类已经实现Comparable接口，会默认调用compareTo方法，故可往TreeSet中添加。
TreeSet <String>ts=new TreeSet<String>();ts.add("a");ts.add("c");ts.add("b");ts.add("2");ts.add("1");System.out.println(ts);
输出结果为[12abc]。
同样的还有诸如Integer、Data类，也默认都实现了Comparable接口。

2 往TreeSet集合中添加自定义对象
当往TreeSet中添加自定义对象实现排序(按自己的方式排序)时，有两种方法。

1>让元素自身具备比较功能：需要实现Comparable接口，覆盖compareTo()方法

2>让集合自身具备比较功能：需要额外定义一个类，实现Comparator接口，重写compare（t1,t2）方法，该类对象作为参数传递给TreeSet集合中的构造函数。

排序规则：先按汽车的名称排序，若相同，按轮子数排序，再相同就不能往里添加了。

第一种方式：

class Car implements Comparable{String name;int viechle;public Car(String name,int viechle){this.name=name;this.viechle=viechle;}public String getNameViechle() {return name+" "+viechle;}public void setNameViechle(String name,int viechle) {this.name = name;this.viechle=viechle;}public int compareTo(Object o){//System.out.println("调用compareTo函数");Car car=(Car)o;int temp=this.name.compareTo(car.name);return temp==0?this.viechle-car.viechle:temp;}
}
---------------------------------------------------------
public class TreeSetDemo {public static void main(String[] args) {// TODO Auto-generated method stubSet <Car>ts=new TreeSet<Car>();ts.add(new Car("c",2));ts.add(new Car("a",3));ts.add(new Car("a",1));ts.add(new Car("b",2));ts.add(new Car("a",1));Iterator<Car> it=ts.iterator();while(it.hasNext()){Car car=it.next();System.out.println(car.getNameViechle());}}
}输出结果：
a 1
a 3
b 2
c 2

第二种方式：

public class comparatorDemo implements Comparator<Car> {@Overridepublic int compare(Car o1, Car o2) {// TODO Auto-generated method stub//System.out.println("调用compare方法");Car c1=(Car)o1;Car c2=(Car)o2;int temp=c1.compareTo(c2);return temp==0?c1.viechle-c2.viechle:temp;}}
----------
public class TreeSetDemo {public static void main(String[] args) {// TODO Auto-generated method stubSet <Car>ts=new TreeSet<Car>(new comparatorDemo());//TreeSet(Comparator<? super E> comparator) ts.add(new Car("c",2));ts.add(new Car("a",3));ts.add(new Car("a",1));ts.add(new Car("b",2));Iterator<Car> it=ts.iterator();while(it.hasNext()){Car car=it.next();System.out.println(car.getNameViechle());}}   
}
输出结果：
a 1
a 3
b 2
c 2