黑马程序员技术交流社区

标题: 对于一个普通的类对象，如何在TreeSet集合中完成排序操作？ [打印本页]
作者: 于汝国    时间: 2011-12-21 13:08
标题: 对于一个普通的类对象，如何在TreeSet集合中完成排序操作？
本帖最后由 于汝国 于 2011-12-21 23:06 编辑

TreeSet的内容是允许进行排序的，但是对于一个普通的类对象是不能向TreeSet集合中加入的，那么、如何在TreeSet集合中实现对普通类对象的排序操作呢？
作者: 海中的游弋草    时间: 2011-12-21 13:10
这是实现自定义treeset排序的一个类代码，请参考一下
package com.shengsiyuan2;

import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;

public class TreeSetTest3
{
        public static void main(String[] args)
        {
                TreeSet set = new TreeSet(new MyComparator());
               
                set.add("C");
                set.add("A");
                set.add("B");
                set.add("E");
                set.add("a");
                set.add("F");
                set.add("D");
               
                for(Iterator iter = set.iterator(); iter.hasNext();)
                {
                        String value = (String)iter.next();
                       
                        System.out.println(value);
                                       
                }
               
        }
}


class MyComparator implements Comparator
{
        public int compare(Object arg0, Object arg1)
        {
                String s1 = (String)arg0;
                String s2 = (String)arg1;
               
                return s2.compareTo(s1);
        }
}
作者: 于汝国    时间: 2011-12-21 13:31
普通类实现了Comparable接口
作者: 海中的游弋草    时间: 2011-12-21 13:32
恩 对呀  
因为 你是对类进行比较的呀
嘿嘿
好好想想 ，好好分析一下代码额  
作者: 刘基军    时间: 2011-12-21 16:32
方法一：元素实现Comparable接口，重写compareTo()方法，实现集合排序
方法二：定义一个比较器传给TreeSet的构造函数，该比较器实现Comparator接口，重写compare()方法，实现集合排序
  1. import java.util.TreeSet;
  2. import java.util.Iterator;
  3. import java.util.Comparator;

  5. /**
  6. 1.实现Comparable接口，重写compareTo()方法，实现集合排序
  7. */
  8. class Student implements Comparable
  9. {
  10.   String name;
  11.   int age;
  12.   Student(String name, int age)
  13.   {
  14.           this.name=name;
  15.           this.age=age;       
  16.   }     
  17.   public int compareTo(Object obj)
  18.   {
  19.                   if(!(obj instanceof Student))
  20.                           throw new RuntimeException("不是学生对象");
  21.                   Student o = (Student)obj;
  22.                   if(this.age==o.age)
  23.                           return this.name.compareTo(o.name);
  24.                   else if(this.age>o.age)
  25.                           return 1;
  26.                   else
  27.                           return -1;
  28.   }
  29.   public String toString()
  30.   {
  31.                   return this.name+"-"+this.age;       
  32.   }
  33. }

  35. class Test
  36. {
  37.         public static void main(String[] args)
  38.         {
  39.                 TreeSet<Student> ts = new TreeSet<Student>();
  40.                 ts.add(new Student("John",20));
  41.                 ts.add(new Student("Jack",20));
  42.                 ts.add(new Student("Jam",21));
  43.                
  44.                 Iterator<Student>it = ts.iterator();
  45.                 while(it.hasNext())
  46.                 {
  47.                         System.out.println(it.next());       
  48.                 }
  49.                
  50.                 /**
  51.                 2.（匿名内部类的方式）实现Comparator接口，重写compare()方法，实现集合排序
  52.                 */
  53.                 TreeSet<Student> ts2 = new TreeSet<Student>(
  54.                 new Comparator<Student>()
  55.                 {
  56.                         public int compare(Student sa, Student sb)
  57.                         {
  58.                                 if(sa.age==sb.age)
  59.                                   return sa.name.compareTo(sb.name);
  60.                           else if(sa.age>sb.age)
  61.                                   return 1;
  62.                           else
  63.                                   return -1;
  64.                         }
  65.                 }
  66.                 );
  67.                
  68.                 ts2.add(new Student("John",20));
  69.                 ts2.add(new Student("Jack",20));
  70.                 ts2.add(new Student("Jam",21));
  71.                 System.out.println();       
  72.                
  73.                 Iterator<Student>it2 = ts2.iterator();
  74.                 while(it2.hasNext())
  75.                 {
  76.                         System.out.println(it2.next());       
  77.                 }
  78.   }
  79. }
复制代码
测试结果：
  1. Jack-20
  2. John-20
  3. Jam-21

  5. Jack-20
  6. John-20
  7. Jam-21
复制代码
作者: 周敏2011nc    时间: 2011-12-21 18:21
TreeSet sortedResults = new TreeSet(
new Comparator(){
public int compare(Object o1, Object o2) {
     // TODO Auto-generated method stub
        User user1 = (User)o1;
        User user2 = (User)o2;
/*如果compareTo返回结果0，则认为两个对象相等，新的对象不
会增加到集合中去
* 所以，不能直接用下面的代码，否则，那些个数相同的其他姓
名就打印不出来。
* */
//return user1.value-user2.value;
//return
        user1.value<user2.value?-1:user1.value==user2.value?0:1;
       if(user1.value<user2.value)
       {
            return -1;
       }else if(user1.value>user2.value) {
          return 1;
      }else{
     return user1.name.compareTo(user2.name);
    }
   }
}
作者: 于汝国    时间: 2011-12-21 23:05
谢谢各位！明白了，嘻嘻！黑马这是个温暖的大家庭！
作者: 于汝国    时间: 2011-12-21 23:05
谢谢各位！明白了，嘻嘻！黑马这是个温暖的大家庭！



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