黑马程序员技术交流社区

标题: TreeSet为学生类按照条件排序 [打印本页]
作者: 小七的    时间: 2015-3-21 19:25
标题: TreeSet为学生类按照条件排序
创建一个学生类，分别声明String name、int math、int chinese三个私有变量，并提供一系列的公共访问方法和一个总分类：getSum，这里就不贴出Student类了，主要看main函数。
代码如下：
  1. import java.util.Comparator;
  2. import java.util.Scanner;
  3. import java.util.TreeSet;

  5. public class Demo {
  6.         public static void main(String[] args) {
  7.                 Scanner sc = new Scanner(System.in);
  8.                 TreeSet<Student> stuSet = new TreeSet<>(
  9.                                
  10.                         new Comparator<Object>(){
  11.                         public int compare(Object o1, Object o2) {
  12.                                 Student stu1 = (Student)o1;
  13.                                 Student stu2 = (Student)o2;
  14.                                 //比较总分：降序
  15.                                 int sum = stu2.getSum() - stu1.getSum();
  16.                                 //如果总分相同，排语文分，升序
  17.                                 int num1 = (sum == 0 ? stu1.getChinese() - stu2.getChinese() : sum);
  18.                                 //如果语文相同，排数学
  19.                                 int num2 = (num1 == 0 ? stu1.getMath() - stu2.getMath() : num1);
  20.                                 return num2;
  21.                         }});
  22.                         for(int i = 1 ;i <= 2 ; i++){
  23.                                 System.out.println("请输入第 " + i + " 名学员信息：");
  24.                                 System.out.print("姓名：");
  25.                                 String name = sc.next();
  26.                                 System.out.print("语文：");
  27.                                 int chinese = sc.nextInt();
  28.                                 System.out.print("数学");
  29.                                 int math = sc.nextInt();
  30.                                
  31.                                
  32.                                 //封装Student对象
  33.                                 Student stu = new Student(name,chinese,math);
  34.                                 //添加到集合
  35.                                 stuSet.add(stu);
  36.                         }
  37.                        
  38.                         System.out.println("遍历集合：");
  39.                         System.out.println("\t姓名\t总分\t语文\t数学");
  40.                         for(Student stu : stuSet){
  41.                                 System.out.println("\t"  + stu.getName() + "\t" + stu.getSum() + "\t" +
  42.                                                                         stu.getChinese() + "\t" + stu.getMath() + "\t" );
  43.                         }
  44.         }
  45. }
复制代码


这种排序的方法，也可以用ArrayList来做，思路相似，ArrayList可以实现接口compareble，重写compreTo方法。



欢迎光临 黑马程序员技术交流社区 (http://bbs.itheima.com/) 黑马程序员IT技术论坛 X3.2