刷题时候遇到treeset的应用,虽然课上没学到,但结合API和百度大概了解了下API的应用,其实Treeset是Sorted接口的实现类,其中支持两种排序方法,自然排序与定制排序,下面以自然排序调用compareTo(Object o)方法实现为例进行描述。
需求: (1)定义一个员工类Employee,包含三个属性:姓名(String name)、年龄(int age)、工资(int salary) (2)创建5个对象装入TreeSet,按照工资从高到底排序输出结果(工资相同,按照年龄从低到高,工资和年龄都相同,按照姓名字典顺序排列,但是不能去重(姓名、年龄、工资都相同)) import java.util.Comparator;
import java.util.Scanner;
import java.util.TreeSet;
import com.heima.bean.employee;
public class linaxi_1 {
public static void main(String[] args) {
TreeSet<employee> tr = new TreeSet<employee>(new Comparator<employee>() {
public int compare(employee e1, employee e2) {
double num = (e2.getSalary()-e1.getSalary());
double age1 = num ==0?e1.getAge()-e2.getAge():num;
return (int) (num ==0? e1.getName().compareTo(e2.getName()):num);
}
});
tr.add(new employee("张三",23,4000));
tr.add(new employee("李四",24,1000));
tr.add(new employee("周七",23,1000));
tr.add(new employee("王五",25,5000));
for (employee e : tr) {
System.out.println(e);
}
}
} |