java Collection中的排序问题
这里讨论list、set、map的排序，包括按照map的value进行排序。
1）list排序
list排序可以直接采用Collections的sort方法，也可以使用Arrays的sort方法，归根结底Collections就是调用Arrays的sort方法。
public static <T> void sort(List<T> list, Comparator<? super T> c) { Object[] a = list.toArray(); Arrays.sort(a, (Comparator)c); ListIterator i = list.listIterator(); for (int j=0; j<a.length; j++) { i.next(); i.set(a[j]); } } |
如果是自定义对象，需要实现Comparable接口使得对象自身就有“比较”的功能，当然我们也可以在外部使用Comparator来规定其排序。
例如：
package com.fox;/** * @author huangfox * @data 2012-7-5 * @email huangfox009@126.com * @desc */public class User implements Comparable<User> { private String name; private int age; public User() { } public User(String name, int age) { super(); this.name = name; this.age = age; } @Override public String toString() { return "name:" + name + ",age:" + age; } public String getName() { return name; } public void setName(String name) { this.name = name; } public int getAge() { return age; } public void setAge(int age) { this.age = age; } @Override public int compareTo(User o) { if (o.age < this.age) return 1; else if (o.age > this.age) return -1; else return 0; } /** * @param args */ public static void main(String[] args) { User u1 = new User("fox", 11); User u2 = new User("fox2", 21); System.out.println(u2.compareTo(u1)); }} |
排序：
// List<User> us = new ArrayList<User>(); List<User> us = new LinkedList<User>(); us.add(new User("f5", 12)); us.add(new User("f2", 22)); us.add(new User("f3", 2)); us.add(new User("f4", 14)); us.add(new User("f5", 32)); us.add(new User("f4", 12)); us.add(new User("f7", 17)); us.add(new User("f8", 52)); System.out.println(us.toString()); long bt = System.nanoTime(); Collections.sort(us, new Comparator<User>() { @Override public int compare(User o1, User o2) { if (o1.getAge() < o2.getAge()) return -1; else if (o1.getAge() > o2.getAge()) return 1; else return o1.getName().compareTo(o2.getName()); } }); long et = System.nanoTime(); System.out.println(et - bt); System.out.println(us.toString()); |
当然这里可以直接Collections.sort(us)，这里用Comparator对User自身的比较方法compareTo做了一点点优化（对相同年龄的人根据用户名排序，String的排序）。
简单提一下，Arrays的排序采用的是插入排序和归并排序，当数组长度较小时直接插入排序。
2）set排序
set包括HashSet和TreeSet，HashSet是基于HashMap的，TreeSet是基于TreeMap的。
TreeMap是用红黑树实现，天然就具有排序功能，“天然就具有排序功能”是指它拥有升序、降序的迭代器。
那么HashSet怎么排序呢？我们可以将HashSet转成List，然后用List进行排序。
例如：
Set<User> us = new HashSet<User>(); // Set<User> us = new TreeSet<User>(); // Set<User> us = new TreeSet<User>(new Comparator<User>() { // // @Override // public int compare(User o1, User o2) { // if (o1.getAge() < o2.getAge()) // return -1; // else if (o1.getAge() > o2.getAge()) // return 1; // else // return o1.getName().compareTo(o2.getName()); // } // }); us.add(new User("f5", 12)); us.add(new User("f2", 22)); us.add(new User("f3", 2)); us.add(new User("f4", 14)); us.add(new User("f5", 32)); us.add(new User("f4", 12)); us.add(new User("f7", 17)); us.add(new User("f8", 52)); // set -> array List<User> list = new ArrayList<User>(us); System.out.println(list); Collections.sort(list); System.out.println(list); |
也可以将HashSet转换成数组用Arrays排序。
3）map排序
map包括HashMap和TreeMap，上面已经提过，TreeMap用红黑树实现，天然具有排序功能。
那么HashMap怎么按“key”排序呢？方法很简单，用HashMap来构造一个TreeMap。
Map<String, Integer> us = new HashMap<String, Integer>(); // Map<String, Integer> us = new TreeMap<String, Integer>(); us.put("f1", 12); us.put("f2", 13); us.put("f5", 22); us.put("f4", 42); us.put("f3", 15); us.put("f8", 21); us.put("f6", 123); us.put("f7", 1); us.put("f9", 19); System.out.println(us.toString()); System.out.println(new TreeMap<String, Integer>(us)); |
怎么按照“value”进行排序？
// 按照value排序 Set<Entry<String, Integer>> ks = us.entrySet(); List<Entry<String, Integer>> list = new ArrayList<Map.Entry<String, Integer>>( ks); Collections.sort(list, new Comparator<Entry<String, Integer>>() { @Override public int compare(Entry<String, Integer> o1, Entry<String, Integer> o2) { if (o1.getValue() < o2.getValue()) return -1; else if (o1.getValue() > o2.getValue()) return 1; return 0; } }); System.out.println(list); |
将map的Entry提出成set结构，然后将set转成list，最后按照list进行排序。
