首页 > 代码库 > TreeSet联系 自然排序

TreeSet联系 自然排序

******************************************************

创建需要的两个类

package com.hu.treeset;

public class MyDate {
private int day;
private int month;
private int year;
public int getDay() {
return day;
}
public void setDay(int day) {
this.day = day;
}
public int getMonth() {
return month;
}
public void setMonth(int month) {
this.month = month;
}
public int getYear() {
return year;
}
public void setYear(int year) {
this.year = year;
}
@Override
public String toString() {
return "MyDate [day=" + day + ", month=" + month + ", year=" + year
+ "]";
}
public MyDate(int day, int month, int year) {
super();
this.day = day;
this.month = month;
this.year = year;
}
@Override
public int hashCode() {
final int prime = 31;
int result = 1;
result = prime * result + day;
result = prime * result + month;
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 (month != other.month)
return false;
if (year != other.year)
return false;
return true;
}

}

**********************************************************

package com.hu.treeset;

public class Employee implements Comparable{
private String name;
private int age;
private MyDate birthDate;
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;
}
public MyDate getBirthDate() {
return birthDate;
}
public void setBirthDate(MyDate birthDate) {
this.birthDate = birthDate;
}
@Override
public String toString() {
return "Employee [name=" + name + ", age=" + age + ", birthDate="
+ birthDate + "]";
}
public Employee(String name, int age, MyDate birthDate) {
super();
this.name = name;
this.age = age;
this.birthDate = birthDate;
}
@Override
public int compareTo(Object o) {
if(o instanceof Employee){
Employee e=(Employee) o;
return this.name.compareTo(e.name);
}
return 0;
}
@Override
public int hashCode() {
final int prime = 31;
int result = 1;
result = prime * result + age;
result = prime * result
+ ((birthDate == null) ? 0 : birthDate.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 (birthDate == null) {
if (other.birthDate != null)
return false;
} else if (!birthDate.equals(other.birthDate))
return false;
if (name == null) {
if (other.name != null)
return false;
} else if (!name.equals(other.name))
return false;
return true;
}

}

***************************************************

Treeset调用

package com.hu.treeset;

import static org.junit.Assert.*;

import java.util.Iterator;

import org.junit.Test;

public class TreeSet {

//用自然排序
@Test
public void test() {
Employee e1=new Employee("MuYongyuan", 12, new MyDate(04, 11, 2001));
Employee e2=new Employee("ZhouYiming", 12, new MyDate(07, 10, 2002));
Employee e3=new Employee("YuXi",9, new MyDate(04, 11, 2004));
Employee e4=new Employee("Wanghaoyu ", 15, new MyDate(04, 11, 2000));
Employee e5=new Employee("zhanganyue", 12, new MyDate(04, 11, 2003));

java.util.TreeSet set=new java.util.TreeSet();
set.add(e1);
set.add(e2);
set.add(e3);
set.add(e4);
set.add(e5);

Iterator iterator=set.iterator();
while(iterator.hasNext()){
System.out.println(iterator.next());
}

}

}

******************************************

定制排序

//用自然排序
@Test
public void test() {
Employee e1=new Employee("MuYongyuan", 12, new MyDate(04, 11, 2001));
Employee e2=new Employee("ZhouYiming", 12, new MyDate(07, 10, 2002));
Employee e3=new Employee("YuXi",9, new MyDate(04, 11, 2004));
Employee e4=new Employee("Wanghaoyu ", 15, new MyDate(04, 11, 2000));
Employee e5=new Employee("zhanganyue", 12, new MyDate(04, 11, 2003));

java.util.TreeSet set=new java.util.TreeSet();
set.add(e1);
set.add(e2);
set.add(e3);
set.add(e4);
set.add(e5);

Iterator iterator=set.iterator();
while(iterator.hasNext()){
System.out.println(iterator.next());
}

}
//定制排序
@Test
public void test2(){

Comparator comparator=new Comparator() {

@Override
public int compare(Object o1, Object o2) {
if(o1 instanceof Employee1 && o2 instanceof Employee1){
Employee1 e1=(Employee1)o1;
Employee1 e2=(Employee1)o2;

MyDate bir1=e1.getBirthDate();
MyDate bir2=e2.getBirthDate();
if(bir1.getYear()!=bir2.getYear()){
return bir1.getYear()-bir2.getYear();

}else{
if(bir1.getMonth()!=bir2.getMonth()){
return bir1.getMonth()-bir2.getMonth();
}
else{
return bir1.getDay()-bir2.getDay();
}

}
}
return 0;
}
};
java.util.TreeSet set=new java.util.TreeSet(comparator);

Employee1 e1=new Employee1("MuYongyuan", 12, new MyDate(04, 10, 2002));
Employee1 e2=new Employee1("ZhouYiming", 12, new MyDate(07, 10, 2002));
Employee1 e3=new Employee1("YuXi",9, new MyDate(04, 11, 2004));
Employee1 e4=new Employee1("Wanghaoyu ", 15, new MyDate(04, 11, 2000));
Employee1 e5=new Employee1("zhanganyue", 12, new MyDate(04, 11, 2003));

set.add(e1);
set.add(e2);
set.add(e3);
set.add(e4);
set.add(e5);

Iterator iterator=set.iterator();
while(iterator.hasNext()){
System.out.println(iterator.next());
}

}

TreeSet联系 自然排序