Go语言视频零基础入门到精通

javalearner

TreeSet基于TreeMap，它很有意思，当你执行add()方法的时候，集合会自动排序，不需要手工执行Collections.sort()进行排序，代码更加简洁(Clean)，今天使用过程发现它暗藏机关，与大家分享。

使用TreeSet有两种方法：
一、元素实现Comparable接口，然后直接通过add方法添加元素即可。
二、创建一个Comparator，并在构造TreeSet的时候传入，如下：


//我这里Job为待添加元素
Set<Job> jobs = new TreeSet<Job>(new JobComparator());

我要实现的需求很简单，根据Job的优先级字段(Priority)进行排序，Job类代码如下：

1. public class Job implements Comparable
2.    
3.       {  
   
4.   
   
5.     private String jobName;  
   
6.   
   
7.     private Integer priority;  
   
8.   
   
9.     public SortJob() {  
   
10.     }  
    
11.   
    
12.     public SortJob(String jobName, Integer priority) {  
    
13.         this.jobName = jobName;  
    
14.         this.priority = priority;  
    
15.     }  
    
16.   
    
17.     public String getJobName() {  
    
18.         return jobName;  
    
19.     }  
    
20.   
    
21.     public void setJobName(String jobName) {  
    
22.         this.jobName = jobName;  
    
23.     }  
    
24.   
    
25.     public Integer getPriority() {  
    
26.         return priority;  
    
27.     }  
    
28.   
    
29.     public void setPriority(Integer priority) {  
    
30.         this.priority = priority;  
    
31.     }  
    
32.   
    
33.     @Override  
    
34.     public int compareTo(SortJob o) {  
    
35.         return priority.compareTo(o.getPriority());  
    
36.     }  
    
37. }  
38.    

复制代码

测试代码如下：

1. @Test  
   
2. public void sortJob() {  
   
3.     Set< SortJob> jobs = new TreeSet< SortJob>();  
   
4.   
   
5.     jobs.add(new SortJob("001", 1));  
   
6.     jobs.add(new SortJob("002", 1));  
   
7.     jobs.add(new SortJob("003", 1));  
   
8.     assertEquals(3, jobs.size());    // ①  
   
9.   
   
10.     jobs.clear();  
    
11.     jobs.add(new SortJob("001", 1));  
    
12.     jobs.add(new SortJob("002", 3));  
    
13.     jobs.add(new SortJob("003", 2));  
    
14.   
    
15.     List< SortJob> listJobs = new ArrayList< SortJob>(jobs);  
    
16.     assertEquals("001", listJobs.get(0).getJobName());  
    
17.     assertEquals("003", listJobs.get(1).getJobName());  
    
18.     assertEquals("002", listJobs.get(2).getJobName());  
    
19. }  

复制代码

测试并没有通过，在行①，这里实际集合size为1，出了什么问题呢？
  soga，原来TreeSet的add方法会先调用Job的compareTo方法判断元素是否重复，因为Priority全部为1，所以TreeSet认为后两个Job是重复的，这样的结果显然不是我们想要的，重新实现compareTo方法如下：

   @Override
  public int compareTo(SortJob o) {
    int compare1 = priority.compareTo(o.getPriority());
    return compare1 == 0 ? this.jobName.compareTo(o.jobName) : compare1;
  }

再次运行测试，OK了。

总结：

   TreeSet通过集合元素的compareTo方法判断元素是否重复，实现compareTo方法既要考虑排序，又要考虑对象是否重复
发现这个陷阱后，发现Collections.sort()还是很不错的选择，只需要考虑排序就可以了。