java treeset 特性_Java基础之：Set——HashSet & TreeSet

Java基础之：Set——HashSet & TreeSet

HashSet

HashSet实现了Set接口(不可以重复元素)，HashSet实际上底层是HashMap(看后面源码以及HashMap)。

HashSet不保证元素是有序的，顺序取决于hash之后，再进行去索引的结果。

HashSet底层机制(hashCode + equals)

HashSet底层是HashMap

添加一个元素时，先得到此元素的hashCode值，对HashCode值进行计算得到索引值。

找到存储数据表table，查看此索引值位置上是否已经存放有元素。

若没有则直接加入，若有则需要调用equals方法进行比较，如果相同则放弃添加(Set接口不允许重复就是这样判断的)。

若在同一个索引位上有多个元素，它们是以链表的形式存放的，当达到一定数量时，链表会自动树化，变为红黑树。

简单案例

package class_HashSet;

import java.util.HashSet;

​

public class ClassTest01{

​

@SuppressWarnings({ "unchecked", "rawtypes" })

public static void main(String[] args) {

HashSet hashSet = new HashSet();

hashSet.add("Hello01");

hashSet.add("Hello02");

hashSet.add("Hello03");

hashSet.add("Hello04");

for(int i = 0;i<8;i++) { //将会有8个Dog对象，放在同一个索引处，形成链表

hashSet.add(new Dog());

}

System.out.println(hashSet);

}

​

}

class Dog{

@Override

public int hashCode() {

//让Dog的所有实例对象的hashCode都为1

return 1;

}

@Override

public boolean equals(Object obj) {

//让Dog的所有实例对象在比较时都返回false，即都不相同。

return false;

}

​

//当hashCode固定为1，equals为false时，会一直加入到HashSet数据表中的一个索引位上

}

实际应用案例

定义一个Employee类，该类包含：private成员属性name(String),age(int),birthday(MyDate)，

其中 birthday 为 MyDate类(属性包括：year(int), month(int),day(int))的对象, 要求:

创建3个Employee 放入 HashSet中，当 name和birthday一样时，认为是同一个员工, 就不能添加到HashSet集合中

提示: 根据前面HashSet 的添加机制，想办法 重写 Employe和MyDate的hashCode 和 equals方法

比如 “张三”, 2000-11-11

package class_HashSet;

import java.util.HashSet;

import java.util.Set;

public class ClassWork01 {

@SuppressWarnings({ "unchecked", "rawtypes" })

public static void main(String[] args) {

Employee employee = new Employee("小范", 20, new MyDate(2000, 3, 26));

Employee employee2 = new Employee("小黄", 18, new MyDate(1999, 11, 25));

Employee employee3 = new Employee("小范", 20, new MyDate(2000, 3, 26));

Set set = new HashSet();

set.add(employee);

set.add(employee2);

set.add(employee3);

for(Object obj : set) {

System.out.println(obj);

}

}

}

class Employee{

private String name;

private int age;

private MyDate birthday;

public Employee(String name, int age, MyDate birthday) {

super();

this.name = name;

this.age = age;

this.birthday = birthday;

}

@Override

public int hashCode() {

final int prime = 31;

int result = 1;

result = prime * result + ((birthday == null) ? 0 : birthday.hashCode());

result = prime * result + ((name == null) ? 0 : name.hashCode());

return result;

}

@Override

public boolean equals(Object obj) {

if (this == obj)

return true;

if (obj == null)

return false;

if (getClass() != obj.getClass())

return false;

Employee other = (Employee) obj;

if (age != other.age)

return false;

if (birthday == null) {

if (other.birthday != null)

return false;

} else if (!birthday.equals(other.birthday))

return false;

if (name == null) {

if (other.name != null)

return false;

} else if (!name.equals(other.name))

return false;

return true;

}

@Override

public String toString() {

return "Employee [name=" + name + ", age=" + age + ", birthday=" + birthday + "]";

}

}

class MyDate{

private int year;

private int mouth;

private int day;

public MyDate(int year, int mouth, int day) {

super();

this.year = year;

this.mouth = mouth;

this.day = day;

}

public int getYear() {

return year;

}

public void setYear(int year) {

this.year = year;

}

public int getMouth() {

return mouth;

}

public void setMouth(int mouth) {

this.mouth = mouth;

}

public int getDay() {

return day;

}

public void setDay(int day) {

this.day = day;

}

@Override

public int hashCode() {

final int prime = 31;

int result = 1;

result = prime * result + day;

result = prime * result + mouth;

result = prime * result + year;

return result;

}

@Override

public boolean equals(Object obj) {

if (this == obj)

return true;

if (obj == null)

return false;

if (getClass() != obj.getClass())

return false;

MyDate other = (MyDate) obj;

if (day != other.day)

return false;

if (mouth != other.mouth)

return false;

if (year != other.year)

return false;

return true;

}

@Override

public String toString() {

return year + "-" + mouth + "-" + day;

}

}

程序输出：

Employee [name=小范, age=20, birthday=2000-3-26]

Employee [name=小黄, age=18, birthday=1999-11-25]

HashSet源码追踪

TreeSet

TreeSet实现了Set接口，所以有Set接口的所有特性。

TreeSet与HashSet不同的是，TreeSet有序的。

TreeSet可以自定义一个Comparator，定义一个排序规则。

package class_TreeSet_LinkedHashSet;

import java.util.Comparator;

import java.util.TreeSet;

public class ClassTest01_TreeSet {

@SuppressWarnings({ "rawtypes", "unchecked" })

public static void main(String[] args) {

TreeSet ts = new TreeSet(new Comparator() {

@Override

public int compare(Object o1, Object o2) {

return ((String)o1).length() - ((String)o2).length();

}

}) ;

//假设ts集合中放的都是字符串，让ts集合中的元素按照长度从小到大排列输出。

ts.add("Hello");

ts.add("小范");

ts.add("aaaaa");

ts.add("xiaofan");

System.out.println(ts);

TreeSet ts1 = new TreeSet();

ts1.add(new Dog());

ts1.add(new Dog());

ts1.add(new Dog());

}

}

class Dog{

}

程序输出：

[小范, Hello, xiaofan]

程序说明：

可以从结果上面看到，"aaaaa"并没有加入到ts集合中。其原因是，由于我们自定义了比较机制，当compare方法返回值为0时，TreeSet集合里的比较机制就认为两个元素是相同的。(所以这里只要长度相同，就是同一个对象)

而下面的ts1集合中，存入的是Dog对象，但Dog对象本身并没有实现Comparable接口。在底层实现TreeSet的去重机制时，需要将即将添加的对象进行强制类型转换为Comparable类型，再使用其compareTo方法进行比较。所以因为Dog类没有实现Comparable接口，这里会报类型转换异常。