Java基础知识之Set

特性

Set中不允许出现重复元素，是根据什么原理呢？答案是：根据equals()方法来区分的。那么如想自定义类对象实例在Set中不重复出现，则需要覆写equals方法了：

这里我们假定自定义类为Person：

package Public;

public class Person{

    private int age;
    private String name;
    public Person() {
        super();
    }
    public Person(int age, String name) {
        super();
        this.age = age;
        this.name = name;
    }
    //同一对象返回true，反之返回false
    @Override
    public boolean equals(Object obj) {
        //若地址相同，肯定是同一个对象
        if(this == obj){
            return true;
        }
        //若要比较的对象不是自定义类对象（这里为Person）,肯定不是同一对象
        //想一想为什么不写为 this.equals(obj)呢？？？
        if(!(obj instanceof Person)){
            return false;
        }
        //运用多态进行转型
        Person p = (Person)obj;
        //比较各个成员变量的值是否相等
        if(p.age == this.age && p.name.equals(this.name)){
            return true;
        }else{
            return false;
        }
    }
    @Override
    public String toString() {
        return "Person [age=" + age + ", name=" + name + "]";
    }
}

到这里可不要忙着去测试，Set是一个接口，它是这样的：public interface Set extends Collection，为其实例化要使用实现它的类，通常用的有TreeSet（有排序功能）、HashSet（根据hash值存储）。

使用TreeSet实例化：

上面说TreeSet有排序功能，这是根据什么实现的呢？答案是：compareTo()方法，就是说我们自定义的类要实现Comparable接口，覆写compareTo()方法：

让我们该造下Person类：

package Public;
//重点在这里
public class Person implements Comparable<Person>{
    private int age;
    private String name;
    public Person() {
        super();
    }
    public Person(int age, String name) {
        super();
        this.age = age;
        this.name = name;
    }
    //重点在这里
    @Override
    public int compareTo(Person o) {
        if(this.age > o.age){
            return 1;
        }else if(this.age < o.age){
            return -1;
        }else{
            return this.name.compareTo(o.name);
        }
    }
    @Override
    public boolean equals(Object obj) {
        if(this == obj){
            return true;
        }
        if(!(obj instanceof Person)){
            return false;
        }
        Person p = (Person)obj;
        if(p.age == this.age && p.name.equals(this.name)){
            return true;
        }else{
            return false;
        }
    }
    @Override
    public String toString() {
        return "Person [age=" + age + ", name=" + name + "]";
    }
}

测试下：

package Set;

import java.util.Set;
import java.util.TreeSet;
import Public.Person;

public class TreeSetDemo {
    public static void main(String[] args) {
        Set<Person> s = new TreeSet<Person>();
        s.add(new Person(3,"x"));//添加元素
        s.add(new Person(1,"x"));//添加元素
        s.add(new Person(1,"x"));//添加重复元素
        s.add(new Person(1,"y"));//添加元素
        s.add(new Person(2,"x"));//添加元素
        System.out.println(s);
    }
}
/*结果：
[Person [age=1, name=x], Person [age=1, name=y], Person [age=2, name=x], Person [age=3, name=x]]*/

从结果看出，添加的元素没有重复，且输出有序；

使用HashSet实例化

HashSet并不具有排序，所以可以不实现Comparable接口，但是要覆写hashCode()方法，毕竟人家要有自己的东西去标示下嘛；

简单的修改下Person类：

package Public;

public class Person{

    private int age;
    private String name;
    public Person() {
        super();
    }
    public Person(int age, String name) {
        super();
        this.age = age;
        this.name = name;
    }
    @Override
    public int hashCode() {
        return this.name.hashCode() * this.age;
    }
    @Override
    public boolean equals(Object obj) {
        if(this == obj){
            return true;
        }
        if(!(obj instanceof Person)){
            return false;
        }
        Person p = (Person)obj;
        if(p.age == this.age && p.name.equals(this.name)){
            return true;
        }else{
            return false;
        }
    }
    @Override
    public String toString() {
        return "Person [age=" + age + ", name=" + name + "]";
    }
}

测试一下：

package Set;

import java.util.HashSet;
import java.util.Set;
import Public.Person;
public class HashSetDemo {
    public static void main(String[] args) {
        Set<Person> s = new HashSet<Person>();
        s.add(new Person(1,"x"));
        s.add(new Person(1,"x"));
        s.add(new Person(1,"z"));
        System.out.println(s);
    }
}
/*结果：
[Person [age=1, name=z], Person [age=1, name=x]]*/

好了，就讲到这里吧！