老紫竹JAVA提高教程(4)-认识Set集合之TreeSet

原文地址：http://www.java2000.net/p11837

TreeSet 是用来进行集合排序的，请注意他和LinkedHashSet的区别。

TreeSet是按照一定的规则，将加入到集合里面的数据进行排序，而LinkedHashSet是严格按照你放入集合的顺序进行使用。
先看源代码

1. package collection.lession4;
3. import java.util.Arrays;
4. import java.util.Comparator;
5. import java.util.Set;
6. import java.util.TreeSet;
8. /**
9.  * 老紫竹JAVA提高教程(4)-认识Set集合之TreeSet。<br>
10.  * 此实现的特点是里面的数据会自然排序或者用指定的排序类排序。
11.  * 
12.  * @author 老紫竹 JAVA世纪网(java2000.net)
13.  * 
14.  */
15. public class Lession4 {
17.   public static void main(String[] args) {
18.     // 测试自然排序
19.     testNature();
21.     // 测试指定排序方式
22.     testComparator();
24.     // 测试特殊的混合对象排序
25.     testMix();
26.   }
28.   public static void testNature() {
29.     // 测试一下数字
30.     TreeSet<Integer> set = new TreeSet<Integer>();
31.     // 反向存入整数数据
32.     for (int i = 10; i >= 1; i--) {
33.       set.add(i);
34.     }
35.     // 输出看看
36.     // 可以看到数据为自然的数字排序
37.     showSet(set); // [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
39.     // 测试一下字符串
40.     TreeSet<String> set2 = new TreeSet<String>();
41.     // 同样反向放入
42.     for (int i = 10; i >= 1; i--) {
43.       set2.add(i + "");
44.     }
45.     // 看输出结果
46.     // 10 的自然排序没有2高，因为字符'1'小于'2'
47.     showSet(set2); // [1, 10, 2, 3, 4, 5, 6, 7, 8, 9]
49.     // TreeSet里面的数据因为要排队
50.     // 所以如果是混合类型数据的数据，如果不能字段转换
51.     // 将出现异常 java.lang.ClassCastException:
52.   }
54.   public static void testComparator() {
55.     // 同样测试数字，我们要反向排序
56.     TreeSet<Integer> set = new TreeSet<Integer>(new MyIntegerComparator());
57.     // 反向存入整数数据
58.     for (int i = 10; i >= 1; i--) {
59.       set.add(i);
60.     }
61.     // 输出看看
62.     // 可以看到数据为我们期望的反向排序了
63.     showSet(set); // [10, 9, 8, 7, 6, 5, 4, 3, 2, 1]
65.     // 我们指定更奇怪的排序，奇数在前面，偶数在后面
66.     // 我们使用匿名内部类
67.     TreeSet<Integer> set2 = new TreeSet<Integer>(new Comparator<Integer>() {
68.       @Override
69.       public int compare(Integer o1, Integer o2) {
70.         if (o1 % 2 != 0) {
71.           if (o2 % 2 != 0) {
72.             return o2.compareTo(o1);
73.           }
74.           return -1;
75.         }
76.         if (o2 % 2 != 0) {
77.           return 1;
78.         }
79.         return o2.compareTo(o1);
80.       }
81.     });
82.     // 反向存入整数数据
83.     for (int i = 10; i >= 1; i--) {
84.       set2.add(i);
85.     }
86.     // 输出看看
87.     // 可以看到数据，技术的在前面，且从大到小排序
88.     // 偶数在后面，也是从大到小排序
89.     showSet(set2); // [9, 7, 5, 3, 1, 10, 8, 6, 4, 2]
91.   }
93.   /**
94.    * 混合对象，你要为每个对象类型进行计较控制，避免出现转换异常.
95.    */
96.   public static void testMix() {
97.     // 我们测试混合类型的对象，比如字符串和整数
98.     // 如果我们不指定自己的比较器，而使用默认的自然比较器
99.     // 将发生异常
100.     TreeSet set = new TreeSet(new Comparator() {
101.       @Override
102.       public int compare(Object o1, Object o2) {
103.         // 我们用字符串比较好了
104.         return o1.toString().compareTo(o2.toString());
105.       }
107.     });
108.     for (int i = 10; i >= 1; i--) {
109.       set.add(i); // 存入整数
110.       set.add(i + ""); // 存入字符串
111.     }
112.     // 输出看看
113.     // 里面的内容确实按照字符串进行排序了。
114.     // 数字被转化为字符串再参与比较。
115.     showSet(set); // [1, 10, 2, 3, 4, 5, 6, 7, 8, 9]
116.   }
118.   /**
119.    * 显示Set里面的数据。
120.    * 
121.    * @param set
122.    */
123.   private static void showSet(Set set) {
124.     System.out.println(Arrays.toString(set.toArray()));
125.   }
126. }
128. // 注意，我指定了强制类型的比较器
129. class MyIntegerComparator implements Comparator<Integer> {
131.   @Override
132.   public int compare(Integer o1, Integer o2) {
133.     return o2.compareTo(o1);// 使用反向的比较
134.   }
136. }

测试：
1 默认排序要求里面的数据类型必须能互相自由转换(实际上就应该是一个类型才行),我们测试如下代码看看运行错误

1.     // 支持混合类型
2.     TreeSet set = new TreeSet();
3.     // 放入混合数据
4.     for (int i = 10; i >= 1; i--) {
5.       set.add(i); // 存入整数
6.       set.add(i + ""); // 存入字符串
7.     }

2 测试让所有的排序都反向，比如以前是从小到大，变成从大到小。 奇数靠前变成偶数考前。


总结：
  当你需要让集合进行有一定排序规则的使用时，记得用这个。 不过我个人更喜欢用List,必要时用sort方法进行排序。