- 向 TreeSet 中添加的元素必须是同一个类的
- 可以按照添加进集合中的元素的指定顺序遍历(仍然是“存储无序”的),如 String、包装类等默认按照从小到大的顺序遍历。
- 当向 TreeSet 中添加自定义类的对象时,有两种排序方法:自然排序、定制排序
- 自然排序:要求该类(要添加进 TreeSet 中的元素所对应的类)实现 java.lang.Comparable 接口(不实现的话不能添加)并实现其 compareTo(Object obj) 方法(非自定义类一般都已经实现了该方法),在该方法中指明按照自定义类的哪个属性进行排序的,该方法返回0时表示两元素用作排序的那个属性是相同的,返回负数代表当前元素的用作排序的那个属性是“字典式地”小于参数元素的用作排序的那个属性的,返回正数代表当前元素的用作排序的那个属性是“字典式地”大于参数元素的用作排序的那个属性的(在设计中首选那个不可能重复的属性来排序,以避免“A属性相同,再按B属性排序”的情况)
- 当向 TreeSet 中添加元素时,首先按照 compareTo( ) 进行比较,一旦返回0,虽然此时仅仅是指定的那个属性一样,但程序会认为两个对象是相同的,后面的对象不能被添加,为避免此情况发生,要求 compareTo( ) 与 equals( ) 和 hashCode( ) 保持一致
- 定制排序:调用TreeSet(Comparator comparator) 构造器构造 TreeSet 的对象,其中 comparator 是 java.util.Comparator 接口的实现类,并且该实现类的 compare(T o1, T o2) 方法指明了排序的根据,此时也要求 compare(T o1, T o2) 与 equals( ) 和 hashCode( ) 保持一致(理由同第5点)
一般不会让一个类即实现 Comparable 又定制 Comparator(这种情况以定制排序为准)
关于两种排序,参考一下代码:
- 自然排序
public class TestTreeSet {
public static void main(String[] args) {
TreeSet t = new TreeSet();
t.add(new Person(12, "b"));
t.add(new Person(12, "d"));
t.add(new Person(11, "c"));
t.add(new Person(10, "b"));
System.out.println(t);// [Person [id=10, name=b],
// Person [id=11, name=c],
// Person [id=12, name=b],
// Person [id=12, name=d]]
}
}
class Person implements Comparable {
private Integer id;
private String name;
public Person(int id, String name) {
super();
this.id = id;
this.name = name;
}
public Person() {
super();
}
public Integer getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
@Override
public int hashCode() {
final int prime = 31;
int result = 1;
result = prime * result + id;
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 (id != other.id)
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 "Person [id=" + id + ", name=" + name + "]";
}
@Override
public int compareTo(Object o) {
if (o instanceof Person) {
Person p = (Person) o;
int i = this.getId().compareTo(p.getId());
if (i == 0) {
return this.getName().compareTo(p.getName());
}
return i;
}
return 0;
}
}
- 定制排序
public class TestTreeSet {
public static void main(String[] args) {
TreeSet t = new TreeSet(new Comparator() {
@Override
public int compare(Object o1, Object o2) {
if (o1 instanceof Person && o2 instanceof Person) {
Person p1 = (Person) o1;
Person p2 = (Person) o2;
int i = p1.getId().compareTo(p2.getId());
if (i == 0) {
return p1.getName().compareTo(p2.getName());
}
return i;
}
return 0;
}
});
t.add(new Person(12, "b"));
t.add(new Person(12, "a"));
t.add(new Person(11, "c"));
t.add(new Person(10, "b"));
System.out.println(t);// [Person [id=10, name=b],
// Person [id=11, name=c],
// Person [id=12, name=a],
// Person [id=12, name=b]]
}
}
class Person {
private Integer id;
private String name;
public Person(int id, String name) {
super();
this.id = id;
this.name = name;
}
public Person() {
super();
}
public Integer getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
@Override
public int hashCode() {
final int prime = 31;
int result = 1;
result = prime * result + id;
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 (id != other.id)
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 "Person [id=" + id + ", name=" + name + "]";
}
}
时间: 2024-10-12 10:25:02