java 有序map_java 有序Map之TreeMap的使用

谦虚使人进步

想要了解一个类，就可以从它实现的接口和继承的父类开始。我们可以看到TreeMap实现了java.util.NavigableMap接口，NavigableMap它又继承了排序Map接口 java.util.SortedMap，因此TreeMap具有排序能力；其次，TreeMap实现了Cloneable和Serializable接口，它也具备克隆和序列化能力

image.png

TreeMap底层由红黑树实现，按照Key的自然顺序升序或者实现Comprator接口进行自定义排序。且TreeMap的排序特性只作用在key上。如果需要value也跟着排序就需要使用一些别的手段。

排序特性

按key排序

默认按key升序排列

package com.springboot.study.demo1;

import java.util.*;

public class Test {

public static void main(String[] args) {

TreeMap treeMap = new TreeMap() {{

put("3", "1");

put("1", "3");

put("2", "2");

}};

for (String s : treeMap.keySet()) {

System.out.println(s+"==>"+treeMap.get(s));

}

}

}

image.png

按key降序排列

package com.springboot.study.demo1;

import java.util.*;

public class Test {

public static void main(String[] args) {

TreeMap treeMap = new TreeMap(new Comparator() {

@Override

public int compare(String o1, String o2) {

return o2.compareTo(o1);

}

}) {{

put("3", "1");

put("1", "2");

put("2", "3");

}};

for (String s : treeMap.keySet()) {

System.out.println(s + "==>" + treeMap.get(s));

}

}

}

image.png

按value排序

如果需要value也跟着排序就需要使用一些别的手段。当然HashMap的value排序也可以使用这种方法

将Map转为 List，然后使用java.util.Collections工具类来排序。当然这种方法也适用于HashMap

package com.springboot.study.demo1;

import java.util.*;

import java.util.stream.Collectors;

public class Test {

public static void main(String[] args) {

Map treeMap = new HashMap() {{

put("3", "1");

put("1", "2");

put("2", "3");

}};

//将Map转为 List

List> list = new ArrayList<>(treeMap.entrySet());

//按照

Collections.sort(list,new Comparator>() {

//升序排序

public int compare(Map.Entry o1, Map.Entry o2) {

return o1.getValue().compareTo(o2.getValue());

}

});

for (Map.Entry e: list) {

System.out.println(e.getKey()+"==>"+e.getValue());

}

}

}

image.png

性能特性

TreeMap的优势在于能够实现自定义排序功能，但是性能要比HashMap和LinkedHashMap差。它的 containsKey 、get 、 put 、remove 方法的时间复杂度是 log(n)

虽然LinkedHashMap也是有序的，但是LinkedHashMap内元素顺序只和插入顺序有关，无法进行自定义排序