Java学习之集合（TreeSet）

TreeSet可以对Set集合中的元素进行排序

判断元素唯一性的方式：就是根据比较方法的返回结果是否为0，是0：就是相同元素，就不会存储

什么比较方法，先不管它，我们先按照以前学过的集合定义和使用

一、自定义元素类型

public class Person {

    private String name;
    private int age;

    public Person() {
    super();
    }

    public Person(String name, int age) {
    super();
    this.name = name;
    this.age = age;
    }

    public String getName() {
    return name;
    }

    public void setName(String name) {
    this.name = name;
    }

    public int getAge() {
    return age;
    }

    public void setAge(int age) {
    this.age = age;
    }

    @Override
    public int hashCode() {
    System.out.println(this + ":-------hashCode");
    return this.name.hashCode() + this.age;
    }

    @Override
    public boolean equals(Object obj) {
    System.out.println(this + ":-------equals-------" + obj);
    if (!(obj instanceof Person))
        throw new ClassCastException("类型不对");
    Person person = (Person) obj;
    return this.name.equals(person.name) && person.age == this.age;
    }

    @Override
    public String toString() {

    return this.name + ":" + this.age;
    }
}

二、业务逻辑代码

import java.util.Iterator;
import java.util.TreeSet;

import cn.marw.common.bean.Person;

public class TreeSetDemo {

    public static void main(String[] args) {
    TreeSet ts = new TreeSet();
    
    ts.add(new Person("zhangsan", 28));
    ts.add(new Person("lisi", 21));
    ts.add(new Person("wangwu", 23));
    ts.add(new Person("zhaoliu", 29));
    ts.add(new Person("zhouqi", 19));
    ts.add(new Person("sunba", 30));
    
    Iterator it = ts.iterator();
    
    while(it.hasNext()) {
        Person p=(Person)it.next();
        System.out.println(p.getName()+":"+p.getAge());
    }
    }

}

运行结果：

 错误信息：Exception in thread "main" java.lang.ClassCastException: class cn.marw.common.bean.Person cannot be cast to class java.lang.Comparable 无法将Person转换为Comparable

首先想到就是修改Person类，这就是TreeSet集合排序的第一种方式（称之为：元素自然顺序）：让元素类型具有比较性（实现Comparable接口，覆盖compareTo方法）：

public class Person implements Comparable {

    private String name;
    private int age;

    public Person() {
    super();
    }

    public Person(String name, int age) {
    super();
    this.name = name;
    this.age = age;
    }

    public String getName() {
    return name;
    }

    public void setName(String name) {
    this.name = name;
    }

    public int getAge() {
    return age;
    }

    public void setAge(int age) {
    this.age = age;
    }

    @Override
    public int hashCode() {
    System.out.println(this + ":-------hashCode");
    return this.name.hashCode() + this.age;
    }

    @Override
    public boolean equals(Object obj) {
    System.out.println(this + ":-------equals-------" + obj);
    if (!(obj instanceof Person))
        throw new ClassCastException("类型不对");
    Person person = (Person) obj;
    return this.name.equals(person.name) && person.age == this.age;
    }

    @Override
    public String toString() {

    return this.name + ":" + this.age;
    }

    @Override
    public int compareTo(Object o) {
    //定义Person类型的自然顺序为：根据年龄排序，年龄相同在根据名称排序
    if(!(o instanceof Person)){
        throw new ClassCastException("类型错误");
    }
    Person p=(Person)o;
    int temp=this.age-p.age;
    return temp==0?this.name.compareTo(p.name):temp;
    }
}

结果这里没有做年龄相同，姓名不能的数据：

这种方式虽然解决了问题，我们想想如果我们修改排序方式那不就影响别人的程序了吗。

第二种方式就来了，让集合具有比较性

具体做法：

1、创建一个类实现Comparator接口，并重写compare方法

2、在实例化TreeSet集合时，把创建的类作为参数传递给TreeSet构造函数

实现Comparator接口的类代码：

import java.util.Comparator;

import cn.marw.common.bean.Person;

public class PersonComparatorByName implements Comparator {

    @Override
    public int compare(Object o1, Object o2) {
    //只比较name
    if(!(o1 instanceof Person)||!(o2 instanceof Person))
        throw new ClassCastException("类型错误");
    
    return ((Person)o1).getName().compareTo(((Person)o2).getName());
    }

}

实例化TreeSet代码

import java.util.Iterator;
import java.util.TreeSet;

import cn.marw.common.bean.Person;

public class TreeSetDemo {

    public static void main(String[] args) {
    TreeSet ts = new TreeSet(new PersonComparatorByName());
    
    ts.add(new Person("zhangsan", 28));
    ts.add(new Person("lisi", 21));
    ts.add(new Person("wangwu", 23));
    ts.add(new Person("zhaoliu", 29));
    ts.add(new Person("zhouqi", 19));
    ts.add(new Person("sunba", 30));
    
    Iterator it = ts.iterator();
    
    while(it.hasNext()) {
        Person p=(Person)it.next();
        System.out.println(p.getName()+":"+p.getAge());
    }
    }
}

结果：