get
首先会判断在初始化TreeMap的时候是否传入了外部选择器Comparator,如果传入了,那么利用Comparator遍历整棵树,找到要存放的位置,利用Comparator的compare()方法比较树根节点p的key和传入的key,compare方法等于0说明相等,那么直接返回根节点,小于0的话p=p.left,大于0的话p=p.right。若没有找到返回null。没有传入Comparator,那么利用key对象实现Comparable接口重写的compareTo()方法,还是先比较根节点p,k.compareTo(p.key)若等于0,返回根节点,小于0的话p=p.left,大于0的话p=p.right。没有找到返回null。
put
首先获取根节点,为空的话,那么直接新建节点,并把当前节点设为根节点。如果不为空,那么获取外部选择器,如果获取到的话,用comparator方式的方式查找put位置,反之利用key的compareTo方法。外部比较器comparator的compare方法允许key值为空,而compareTo方法不允许为null。所以注意了,TreeMap是可以放null的key,但是get(null)却得不到想要的value。遍历整个树如果找到了key相同的节点,那么就把该节点的value值更新为我们的新值。如果遍历完整个树都没找到相同的key值,那么,用key,value新建一个节点,插入到上述遍历过程遍历到的位置。最后插入之后要做平衡处理。