红黑树在之前的文章中已经有说明了https://blog.csdn.net/qq_33605294/article/details/80646098和https://blog.csdn.net/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);