TreeSet的使用

package com.akaiqi.chun.collection;

import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
import java.util.TreeSet;

/**
 * TreeSet
 *
 * @author akaiqi
 * @date 2019/11/4 20:11
 */
public class TreeSetTest {

    /**
     * 4种构造器
     * add:添加一个元素
     * addAll：添加多个元素
     */
    private static void test1() {
        TreeSet<Integer> treeSet1 = new TreeSet<>();
        treeSet1.add(1);
        treeSet1.forEach(System.out::print);//1

        System.out.println("\n");
        List<Integer> list = Arrays.asList(1, 2, 3, 4, 5, 6, 4, 5);
        TreeSet<Integer> treeSet2 = new TreeSet<>(list);
        treeSet2.forEach(System.out::print);//123456

        System.out.println("\n");
        TreeSet<Integer> treeSet3 = new TreeSet<>((Comparator.reverseOrder()));
        treeSet3.addAll(list);
        treeSet3.forEach(System.out::print);//654321

        System.out.println("\n");
        TreeSet<Integer> treeSet4 = new TreeSet<>(treeSet3);
        treeSet4.addAll(list);
        treeSet4.forEach(System.out::print);//654321
    }

    /**
     * ceiling(E e) : 返回此集合中最小元素大于或等于给定元素
     * first() :返回此集合中当前的第一个（最低）元素。
     * floor(E e):返回此集合中最大的元素小于或等于给定元素
     * last():最后一个元素
     * <p>
     * higher(E e) : 返回严格大于给定元素的该集合中的最小元素,没有返回null
     * lower(E e) 返回这个集合中最大的元素严格小于给定的元素，如果没有这样的元素，则返回 null 。
     */
    private static void test2() {
        List<Integer> list = Arrays.asList(0, 1, 2, 3, 4, 5, 6, 7);
        TreeSet<Integer> treeSet = new TreeSet<>(list);
        System.out.println(treeSet.ceiling(3));//3

        System.out.println(treeSet.first());//0
        System.out.println(treeSet.floor(3));//3

        System.out.println(treeSet.last());//7

        System.out.println(treeSet.higher(7));//null
        System.out.println(treeSet.higher(6));//7

        System.out.println(treeSet.lower(0));//null
        System.out.println(treeSet.lower(1));//0
    }

    /**
     * 删除
     * remove(Object o) 从该集合中删除指定的元素
     * pollFirst() 检索并删除最后一个（最高）元素，如果此集合为空，则返回 null 。
     * pollLast() 检索并删除最后一个（最高）元素，如果此集合为空，则返回 null 。
     *
     * clear() :删除所有
     */
    private static void test3() {
        List<Integer> list = Arrays.asList(0, 1, 2, 3, 4, 5, 6, 7);
        TreeSet<Integer> treeSet = new TreeSet<>(list);

        System.out.println(treeSet.remove(0));//true
        treeSet.forEach(System.out::print);//1234567
        System.out.println("\n");

        System.out.println(treeSet.pollFirst());//1
        treeSet.forEach(System.out::print);//234567
        System.out.println("\n");

        System.out.println(treeSet.pollLast());//7
        treeSet.forEach(System.out::print);//23456
        System.out.println("\n");

        treeSet.clear();
    }

    /**
     * 截取部分
     *subSet(E fromElement, boolean fromInclusive, E toElement, boolean toInclusive)
     *  返回该集合的部分的视图，其元素的范围从 fromElement到 toElement 。
     *
     *  subSet(E fromElement, E toElement) :（a，b）
     *
     *  tailSet(E fromElement) [a,tail]
     *
     *  tailSet(E fromElement, boolean inclusive)
     */
    private static void test4() {
        List<Integer> list = Arrays.asList(0, 1, 2, 3, 4, 5, 6, 7);
        TreeSet<Integer> treeSet = new TreeSet<>(list);

        System.out.println(treeSet.subSet(1,true,6,true));//[1, 2, 3, 4, 5, 6]

        System.out.println(treeSet.subSet(2,5));//[2, 3, 4]

        System.out.println(treeSet.tailSet(2));//[2, 3, 4, 5, 6, 7]
    }


    public static void main(String[] args) {
        test4();
    }
}