package com.test;
import java.util.TreeSet;
public class Demo8 {
public static void main(String[] args) {
/*
* TreeSet:可以实现去重和排序
*/
TreeSet set = new TreeSet<>();
/*
* 使用TreeSet在存储字符串的时候,自动实现了去重顺便实现了排序
* 在add()方法中调用了字符串的int compareTo(Object obj)方法
* 字符串类实现了Comparable接口, compareTo(Object obj)方法就来自于这个接口.
*/
//这里默认是按照字典顺序升序排列的
set.add("java1");
set.add("java2");
set.add("java3");
set.add("java3");
set.add("java4");
set.add("java5");
System.out.println(set);//[java1, java2, java3, java4, java5]
//实例:将Person2的对象存入TreeSet实现排序和去重
//分析:让Person2实现Comparable接口的compareTo方法
//比较的规则:按照人的年龄和姓名比较.只要年龄和姓名一致就认为是一个人.
TreeSet set1 = new TreeSet<>();
set1.add(new Person2("bingbing1", 20));
set1.add(new Person2("bingbing2", 200));
set1.add(new Person2("bingbing1", 20));
set1.add(new Person2("bingbing0", 203));
System.out.println(set1);
}
}
class Person2 implements Comparable{
String name;
int age;
public Person2(String name, int age) {
super();
this.name = name;
this.age = age;
}
@Override
public String toString() {
return "Person2 [name=" + name + ", age=" + age + "]";
}
//重写compareTo方法
//按照人的年龄和姓名比较.只要年龄和姓名一致就认为是一个人.
public int compareTo(Object arg0) {
if (!(arg0 instanceof Person2)) {
throw new ClassCastException("类型转换异常");
}
Person2 person2 = (Person2)arg0;
//先按照年龄比,年龄相同再按照姓名比.
int num = age-person2.age;
return num==0?name.compareTo(person2.name):num;
}
}