Key-Value Pair 排序 —— 兼容所有类型

科研中,发现经常要对<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]]

你可能感兴趣的:(java,sort)