Java中的TreeSet

1.TreeSet介绍

TreeSet是一个可以对元素进行排序的容器。HashSet不支持对元素进行排序处理。底层实际是用TreeMap实现的，内部维持了一个简化版的TreeMap，通过Key来存储Set的元素。TreeSet内部需要对存储的元素进行排序。因此，需要给定排序规则。

排序规则实现方式：

• 通过元素自身实现比较规则

• 通过比较器指定比较规则

2.TreeSet使用

import java.util.*;
public class TreeSetTest {
	public static void main(String[] args) {
		//实例化TreeSet
		Set<String> set = new TreeSet<>();
		//添加元素
		set.add("c");
		set.add("a");
		set.add("b");
		set.add("a");
		//获取元素,在这里String规则自身实现了排序规则
		for(String str:set) {
			System.out.println(str);
		}
	}
}	
//重复元素不能出现，做了排序处理
/*
a
b
c
*/

3.通过元素自身实现比较规则

在元素自身实现比较规则时，需要实现Comparable接口中的compareTo方法，该方法中用来定义比较规则。TreeSet通过调用该方法来完成对元素的排序处理。

如果没有定义自身的比较接口，也没有给定外部的比较器，直接使用TreeSet会报错。

//定义Users类
public class Users implements Comparable<Users> {
	private String username;
	private int userage;
	public Users(String username,int userage) {
		this.username = username;
		this.userage = userage;
	}
	public Users() {
		
	}
	public void setUsername(String username) {
		this.username=username;
	}
	public String getUsername() {
		return username;
	}
	public void setUserage(int userage) {
		this.userage = userage;
	}
	public int getUserage() {
		return userage;
	}
	@Override
	public String toString() {
		return "Users{"+"username='"+username+'\''+",userage="+userage+'}';
	}
	@Override 
	public boolean equals(Object o){
		if(this == o ) return true;
		if(o==null||getClass()!=o.getClass()) return false;
		Users users = (Users) o;
		if(userage != users.userage) return false;
		return username != null ? username.equals(users.username):true;
		
	}
	@Override
	public int hashCode() {
		int result = username!=null ? username.hashCode():0;
		result = 31*result+userage;
		return result;
		
	}
	//定义比较规则
	//正数：大，负数：小 0 相等
	@Override
	public int compareTo(Users o) {
		if(this.userage>o.getUserage()) {
			return 1;
		}
		if(this.userage==o.getUserage()) {
			
		}
		return -1;
	}
}

import java.util.*;
public class TreeSetTest02 {
	public static void main(String[] args) {
		Set<Users> set = new TreeSet<>();
		Users u1 = new Users("li1",3);
		Users u2 = new Users("li2",18);
		Users u3 = new Users("li1.5",18);
		set.add(u1);
		set.add(u2);
		set.add(u3);
		for(Users str:set) {
			System.out.println(str);
		}
	}
}
/*
Users{username='li1',userage=3}
Users{username='li1.5',userage=18}
Users{username='li2',userage=18}
*/

4.通过比较器实现比较规则

通过比较器定义比较规则时，需要单独创建一个比较器，比较器需要实现Comparator接口中的compare方法来定义比较规则。在实例化TreeSet时将比较器对象交给TreeSet来完成元素的排序处理。此时元素自身就不需要实现比较规则了。

//创建比较器
import java.util.*;
public class StudentComparator implements Comparator<Student> {
	//定义比较规则
	@Override
	public int compare(Student o1,Student o2) {
		if(o1.getAge()>o2.getAge()) {
			return 1;
		}
		if(o1.getAge()==o2.getAge()) {
			return o1.getName().compareTo(o2.getName());
		}
		return -1;
		
	}
	

}

//创建Student类

public class Student {
	private String name;
	private int age;
	public Student(String name,int age) {
		this.name = name;
		this.age = age;
	}
	public Student(){
		
	}
	
	public void setName(String name) {
		this.name = name;
	}
	public void setAge(int age) {
		this.age = age;
	}
	public String getName() {
		return name;
	}
	public int getAge() {
		return age;
	}
	@Override
	public int hashCode() {
		int result = name!=null ? name.hashCode():0;
		result = 31*result+age;
		return result;
	}
	@Override
	public boolean equals(Object o) {
		if(this ==o) return true;
		if(o == null||getClass()!=o.getClass()) return false;
		Student student = (Student) o;
		if(age!=student.age) return false;
		return name != null ? name.equals(student.name):true;
		
	}
	@Override
	public String toString() {
		return "Student{"+"name='"+name+'\''+",age="+age+'}';
	}
	
		
}

//测试
import java.util.*;
public class TreeSetTest03 {
	public static void main(String[] args) {
		//直接将比较器通过构造方法传给TreeSet
		Set<Student> set=new TreeSet<>(new StudentComparator());
		Student s1 = new Student("li1",3);
		Student s2 = new Student("li2",18);
		Student s3 = new Student("li1.5",18);
		set.add(s1);
		set.add(s2);
		set.add(s3);
		for(Student str:set) {
			System.out.println(str.toString());
		}
	}
}
/*
Student{name='li1',age=3}
Student{name='li1.5',age=18}
Student{name='li2',age=18}
*/