Java——TreeMap和TreeSet用法（基于红黑搜索树的结构）

TreeMap与红黑树详解

最新推荐文章于 2024-08-13 22:44:02 发布

原创最新推荐文章于 2024-08-13 22:44:02 发布 · 324 阅读

0 ·

CC 4.0 BY-SA版权

本文深入解析了TreeMap的使用方法，包括不同类型的遍历方式，如通过keySet、entrySet和values集合的迭代器遍历。同时，文章介绍了如何自定义比较器以改变TreeMap的默认排序，并探讨了基于红黑树结构的TreeSet的基本操作。

红黑树在之前的文章中已经有说明了https://blog.youkuaiyun.com/qq_33605294/article/details/80646098和https://blog.youkuaiyun.com/qq_33605294/article/details/80684672。

本文主要是总结TreeMap的用法，几种遍历的方法，基本都是利用迭代器来做，用迭代器遍历keySet，entrySet，还有Collection的用法

public class test_TreeMap {

	public static void main(String[] args) {
		TreeMap<Integer,String> t=new TreeMap<Integer,String>(new xbComparator());
		//TreeMap<Integer,String> t=new TreeMap<Integer,String>();
		t.put(1,"value_1");
		t.put(2,"value_2");
		t.put(3,"value_3");
		Set<Integer> keys=t.keySet();
		Iterator<Integer> it=keys.iterator();
		while(it.hasNext()) {
			int k=it.next();
			System.out.println(k+" "+t.get(k));
		}
		
		Iterator it1=t.entrySet().iterator();
		while(it1.hasNext()) {
			//注意iterator的强制转换
			Map.Entry<Integer,String> e=(Map.Entry<Integer, String>)it1.next();
			int key=e.getKey();
			String value=e.getValue();
			System.out.println(key+" "+value);
		}
		
		Collection c=t.values();
		Iterator it2=c.iterator();
		while(it2.hasNext()) {
			String value=(String)it2.next();//必须要进行强制转换
		}
		//取红黑树的根节点
		Map.Entry<Integer, String> e=t.firstEntry();
		
		// TODO Auto-generated method stub

	}

}
class xbComparator implements Comparator{
	public int compare(Object o1,Object o2) {
		Integer i1=(Integer)o1;
		Integer i2=(Integer)o2;
		return -i1.compareTo(i2);
	}
	
}

TreeMap中默认的排序为升序，如果要改变其排序可以自己写一个Comparator，还要注意强制转换

还有TreeSet也是基于红黑树的结构，与TreeMap差不太多，ceiling，last，first，代码如下，ceiling，lower，floor都很形象。

lower不包括相等，floor包括相等。

		TreeSet<Integer> ts=new TreeSet<Integer>();
		int size=ts.size();
		ArrayList<Integer> a=new ArrayList<Integer>();
		a.addAll(ts);
		int last=ts.last();
		int first=ts.first();
		ts.pollFirst();
		ts.pollLast();
		int i=1;
		int lower_i=ts.lower(i);
		int j=10;
		int floor_j=ts.floor(j);
		int ceil=ts.ceiling(i);