【笔记】TreeSet的自然排序和定制排序的使用区别

TreeSet特点：

注意：TreeSet满足Set的特点
1.不能重复存储同一个对象，数据储存是以“红黑树”数据结构存储。也是一个平衡二叉树
2.跟Set不同的是以排序方式输出数据

TreeSet的自然排序

步骤1：存储的类要实现Comparable接口
步骤2：重写hashcode（）方法和equals（）方法

示例说明

先创建一个R.java实现comparable比较器类

public class R implements Comparable<R>{
    private int count;

    public R(int count) {
        this.count = count;
    }

    public int getCount() {
        return count;
    }

    public void setCount(int count) {
        this.count = count;
    }

    @Override
    public String toString() {
        return "R{" +
                "count=" + count +
                '}';
    }

    /**
     * 比较两个对象是否相等：
     * 1，hashcode是否相同
     * 2.equals是否是true
     * @return
     */
    @Override
    public int hashCode() {
        return this.getCount();
//        return 1;
    }

    @Override
    public boolean equals(Object obj) {
        if (obj == this){
            return true;
        }else if (obj == null){
            return false;
        }else if (obj instanceof R){
            R otherobj = (R)obj;

            if (otherobj.getCount() == ((R) obj).getCount()){
                return true;
            }
        }
        return false;
//        return true;
    }

    /**
     * 此方法的返回值决定大小
     * 若返回值》0，则当前this比该object小
     * 若返回值《0，则当前this比该object大
     * @param o
     * @return
     */
    @Override
    public int compareTo(R o) {
        return this.getCount()>= o.getCount()?1:-1;
    }

}

新建TreeSetComparable.java

public class TreeSetComparable {
    public static void main(String[] args){
        /**
         * 创建一个TreeSet集合对象
         * 类型要实现Comparable接口
         */
        TreeSet<R> set = new TreeSet<>();
        set.add(new R(456));
        set.add(new R(26));
        set.add(new R(156));
        set.add(new R(356));

        set.forEach((s)->System.out.println(s.toString()));

    }
}

运行结果：

TreeSet的定制排序

定制排序定义

tip：这里用的R.java跟上面是同一个文件，这里就不重复放图了

注意：定制排序要实现Comparator接口，可以看出此接口 是函数接口，只有一个抽象方法

接口方法，返回值是int （1 前者 > 后者 /// -1 前者 < 后者 /// 0 前者 == 后者）

示例说明

新建：TreeSetMainTest.java

import demo01.R;

public class TreeSetMainTest {

	public static <T> void main(String[] args) {
		// TODO Auto-generated method stub
		//定制排序
		TreeSet<R> set = new TreeSet<>((a,b)->a.getCount()>b.getCount() ? -1:1);
		set.add(new R(123));
		set.add(new R(13));
		set.add(new R(132));
		
		//以lambda表达式写法
		set.forEach((v)->System.out.println(v.toString()));
		System.out.println("***************************");
		//以匿名内部类写法
		
		TreeSet<R> set2 = new TreeSet<>(new Comparator<R>() {

			@Override
			public int compare(R o1, R o2) {
				// TODO Auto-generated method stub
				if (o1.getCount() >= o2.getCount()) {
					return 1;
				}else if (o1.getCount() == o2.getCount()) {
					return 0;
				}else {
					return -1;
				}
			}
		});
		
		set2.add(new R(123));
		set2.add(new R(456));
		set2.add(new R(789));
		set2.add(new R(654));
		
		set2.forEach((v)->System.out.println(v));
	}

}

运行结果

总结

java类可以通过实现comparator接口自定义该类比较大小的方法，从而达到定制排序