修改 TreeSet/TreeMap 默认排序规则注意！！

最新推荐文章于 2024-08-07 23:42:58 发布

weixin_34013044

最新推荐文章于 2024-08-07 23:42:58 发布

阅读量781

点赞数

文章标签： python 数据结构与算法

原文链接：https://my.oschina.net/u/1388024/blog/877062

版权

2019独角兽企业重金招聘Python工程师标准>>>

treeMap 基于红黑树实现，treeSet基于treeMap实现，

当我们修改默认的排序规则时候，必须有0的返回，否则可能删除不掉元素

public class TreeSet01 {
    public static void main(String[] args) {
        TreeSet<Integer> kminset = new TreeSet<Integer>();
        kminset.add(1);
        kminset.add(2);
        kminset.remove(kminset.first().intValue());
        System.out.println("kminset size = [" + kminset.size() + "]"); // 1

        TreeSet<Integer> kminset0 = new TreeSet<Integer>(new MyComparator());
        kminset0.add(1);
        kminset0.add(2);
       kminset0.remove(kminset0.first().intValue());
        System.out.println("kminset0 size = [" + kminset0.size() + "]"); //2
    }
    //必须有 0， >0, <0 三个值得返回
    static class MyComparator implements Comparator<Integer> {
        public int compare(Integer f1,Integer f2) {
            if(f1.intValue()== f2.intValue()){
                return 0;
            }
            if (f1.intValue()> f2.intValue())
            {
                return 1;
            }
            return -1;
        }
    }
}

转载于:https://my.oschina.net/u/1388024/blog/877062