文章目录
- 1. TreeSet
- 1.1. 自然排序
- 1.2. 定制排序
1. TreeSet
TreeSet 是 SortedSet 接口的实现类, TreeSet 可以确保集合元素处于排序状态。
TreeSet 支持两种排序方法:自然排序和定制排序。默认情况下, TreeSet 采用自然排序。
1.1. 自然排序
默认情况下, TreeSet 采用自然排序。自然排序: TreeSet 会调用集合元素的 compareTo(Object obj) 方法
来比较元素之间的大小关系,然后将集合元素按升序排列:
• 如果 this > obj, 返回正数 1
• 如果 this < obj, 返回负数 -1
• 如果 this = obj, 返回 0 ,则认为这两个对象相等
• 必须放入同样类的对象 ,( 默认会进行排序 ) 否则可能会发生类型转换异常 ,可以使用泛型来进行限制。
举例:
package day10;
import java.util.Set;
import java.util.TreeSet;
public class Test4 {
public static void main(String[] args) {
Set<Integer> set = new TreeSet<Integer>();
//TreeSet自然排序
set.add(5);
set.add(2);
set.add(4);
set.add(3);
System.out.println(set);
}
}
运行结果:
举例2:实现其他功能
package day10;
import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;
public class Test4 {
public static void main(String[] args) {
Set<Integer> set = new TreeSet<Integer>();
set.add(5);
set.add(2);
set.add(4);
set.add(3);
System.out.println(set);
set.remove(5);
set.contains(3);
// set.clear();//清空集合
//使用迭代器遍历集合
Iterator<Integer> it = set.iterator();
while(it.hasNext()) {
System.out.println(it.next());
}
//for each迭代集合,推荐这种
for(Integer i : set) {
System.out.println(i);
}
}
}
1.2. 定制排序
如果需要实现定制排序,则需要在创建 TreeSet 集合对象时,提供一个 Comparator 接口
的实现类对象。由该 Comparator对象负责集合元素的排序逻辑。
举例:
package day10;
import java.util.Comparator;
import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;
public class Test4 {
public static void main(String[] args) {
Person p1 = new Person("张三", 23);
Person p2 = new Person("李四", 20);
Person p3 = new Person("王五", 16);
Person p4 = new Person("杨六", 29);
Set<Person> set = new TreeSet<Person>(new Person());
set.add(p1);
set.add(p2);
set.add(p3);
set.add(p4);
//for each迭代集合
for(Person p : set) {
System.out.println(p.name + " " + p.age);
}
}
}
class Person implements Comparator<Person>{//把person对象存到TreeSet中并按照年龄排序
int age;
String name;
public Person() {
}
public Person(String name, int age) {
this.name = name;
this.age = age;
}
@Override
public int compare(Person o1, Person o2) {//年龄正序排列
if(o1.age > o2.age) {
return 1;
}else if(o1.age < o2.age) {
return -1;
}else {
return 0;
}
}
}
运行结果: