科研中,发现经常要对<key,value>键值对进行排序,按key(或value)的正序或逆序排序。下面写一简单工具类。
1. 首先定义KeyValuePair类,K和V都采用泛型,可兼容任何类型,以便日后使用。
class KeyValuePair<K extends Comparable<K>, V extends Comparable<V>> implements
Comparable<KeyValuePair<K, V>> {
K key;
V value;
public KeyValuePair(K key, V value) {
this.key = key;
this.value = value;
}
// 此处可定义按照key或value的正序或逆序排序
@Override
public int compareTo(KeyValuePair<K, V> o) {
return -value.compareTo(o.getValue());
}
/**
* @return the key
*/
public K getKey() {
return key;
}
/**
* @return the value
*/
public V getValue() {
return value;
}
@Override
public String toString() {
return "[" + key + "," + value + "]";
}
}
2. 用TreeSet的红黑树原理对KeyValuePair进行排序(此处可换其他排序方法)。下面以key为String类型,value为Integer类型,按照value的逆序进行排序。按照代码如下:
public class KeyValuePairSort {
public static void main(String[] args) {
SortedSet<KeyValuePair<String, Integer>> ts = new TreeSet<KeyValuePair<String, Integer>>();
ts.add(new KeyValuePair<String, Integer>("a", 3));
ts.add(new KeyValuePair<String, Integer>("c", 6));
ts.add(new KeyValuePair<String, Integer>("b", 2));
System.out.println(ts);
}
}
3. 运行输出如下:
[[c,6], [a,3], [b,2]]