使用TreeSet存储Integer对象
TreeSet的特点是可以对存放进去的元素进行排序。
package com.monkey1024.set;
import java.util.TreeSet;
/**
* TreeSet可以对元素进行排序
* 使用TreeSet存储Integer对象
* @author Administrator
*
*/
public class TreeSetTest01 {
public static void main(String[] args) {
TreeSet<Integer> ts = new TreeSet<>();
ts.add(3);
ts.add(3);
ts.add(3);
ts.add(1);
ts.add(1);
ts.add(5);
ts.add(2);
ts.add(4);
ts.add(2);
System.out.println(ts);
}
}
使用TreeSet存储自定义类型对象
这里还是存储之前定义的Person对象，需要实现Comparable接口并且重写compareTo方法，先根据name的字典顺序排序，然后再根据年龄进行排序。
package com.monkey1024.bean;
/**
* Person类
*
*/
public class Person implements Comparable<Person>{
private String name;
private int age;
public Person() {
}
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;
}
// 重写toString方法
@Override
public String toString() {
return "Person [name=" + name + ", age=" + age + "]";
}
@Override
public int hashCode() {
final int prime = 31;
int result = 1;
result = prime * result + age;
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;
Person other = (Person) obj;
if (age != other.age)
return false;
if (name == null) {
if (other.name != null)
return false;
} else if (!name.equals(other.name))
return false;
return true;
}
@Override
public int compareTo(Person o) {
int num = this.name.compareTo(o.getName());
if(num != 0){
//先用name排序
return num;
}else{
//如果姓名排序一样则以age进行排序
return this.age - o.getAge();
}
}
}
测试类：
package com.monkey1024.set;
import java.util.TreeSet;
import com.monkey1024.bean.Person;
/**
* TreeSet可以对元素进行排序
* 使用TreeSet存储Person对象
*
*/
public class TreeSetTest02 {
public static void main(String[] args) {
TreeSet<Person> ts = new TreeSet<>();
ts.add(new Person("tony", 23));
ts.add(new Person("paul", 20));
ts.add(new Person("andy", 22));
ts.add(new Person("james", 24));
ts.add(new Person("lucy", 21));
System.out.println(ts);
}
}
测试类里面如果name写的是中文的话则不能进行汉语拼音排序，因为String类中重写的compareTo方法是根据char类型对应的ascii值进行排序的。
compareTo方法的返回值
TreeSet使用了二叉树的数据结构，负数放到左边，正数放到右边。
当compareTo方法返回0的时候，系统会认为两者一致，所以不会向集合中添加元素
当compareTo方法返回正数的时候，系统将元素存储到右边，所以集合存取顺序一致
compare返回正数
当compareTo方法返回负数的时候，系统将元素存储到左边，所以集合会倒序存储
compare返回负数
Comparator比较器
可以向TreeSet的构造方法中传入一个定制的Comparator，如果传入了Comparator的子类对象, 那么TreeSet就会按照传入对象中定制的规则排序。例如将Person中name按照长度排序。
定制一个CompareByNameLength类实现Comparator接口并重写compare方法，在方法里面编写排序规则
package com.monkey1024.set;
import java.util.Comparator;
import com.monkey1024.bean.Person;
public class CompareByNameLength implements Comparator<Person> {
@Override
public int compare(Person p1, Person p2) {
//根据name长度进行排序
int num = p1.getName().length() - p2.getName().length();
if(num == 0){
num = p1.getName().compareTo(p2.getName());
if(num == 0){
num = p1.getAge() - p2.getAge();
}
}
return num;
}
}
测试类，向TreeSet的构造方法中传入CompareByNameLength对象
package com.monkey1024.set;
import java.util.TreeSet;
import com.monkey1024.bean.Person;
/**
* TreeSet可以对元素进行排序
* 使用TreeSet根据name长度进行排序
*
*/
public class TreeSetTest03 {
public static void main(String[] args) {
TreeSet<Person> ts = new TreeSet<>(new CompareByNameLength());
ts.add(new Person("tony", 23));
ts.add(new Person("jordan", 20));
ts.add(new Person("tim", 22));
ts.add(new Person("james", 24));
ts.add(new Person("an", 21));
System.out.println(ts);
}
}
练习
1.将一个字符串中的字符按照字典顺序进行排序，要保留重复的字符。例如输入：java，打印aajv
答案
1.向输入的字符串转换为字符数组，然后向TreeSet中添加字符
package com.monkey1024.set.exercise;
import java.util.Comparator;
import java.util.Scanner;
import java.util.TreeSet;
/**
* 问题：将一个字符串中的字符按照字典顺序进行排序，要保留重复的字符。例如输入：java，打印aajv
* 分析：向输入的字符串转换为字符数组，然后向TreeSet中添加字符
*/
public class Exercise03 {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in); //创建键盘录入对象
System.out.println("请输入一行字符串:");
String line = sc.nextLine(); //将键盘录入的字符串存储在line中
char[] arr = line.toCharArray(); //将字符串转换成字符数组
TreeSet<Character> ts = new TreeSet<>(new Comparator<Character>() {
@Override
public int compare(Character c1, Character c2) {
int num = c1.compareTo(c2);
//保存重复数据
return num == 0 ? 1 : num;
}
});
for(char c : arr) {
ts.add(c);
}
for(Character ch : ts) {
System.out.print(ch);
}
}
}