TreeSet实现Comparator接口的排序算法的分析
为了方便,用lambda表达式代替comparator接口
例子如下:
public static void main(String[] args) { TreeSet<Integer> nums = new TreeSet<>((a,b) -> (2*b-a)); nums.add(5); nums.add(2); nums.add(6); nums.add(-4); System.out.println(nums); } =============输出============== [5, 6, 2, -4]
猜测,使用的是二叉排序树(红黑树)。
猜测,a是刚刚要插入的元素,b是老元素
猜测,如果比较表达式(2*b-a)大于0,那么a插入到b的右边;如果(2*b-a)小于0,那么a插入到b的左边;如果(2*b-a)等于0,那么a不插入(set不可有重复元素)。
所以,
2插入到5的右边,
6插入到5的右边,2的左边
-4插入到5的右边,2的右边
插入完成,二叉树如下(画图软件是猫头鹰,可截图可画图,很好用),
打印时,按照每科树的从左往右(从根节点开始,先打印左子树,再打印根节点,再打印右子树)的递归顺序,
打印结果为5,6,2,-4