EnumSet的使用

应用场景一

在一些工作中(如医生、客服),不是每个工作人员每天都在的,每个人可工作的时间是不一样的,比如张三可能是周一和周三,李四可能是周四和周六,给定每个人可工作的时间,我们可能有一些问题需要回答。比如:
❑ 有没有哪天一个人都不会来?
❑ 有哪些天至少会有一个人来?
❑ 有哪些天至少会有两个人来?
❑ 有哪些天所有人都会来,以便开会?
❑ 哪些人周一和周二都会来?(案例引用自书籍)

实现

先有一个表示星期的日期枚举类

enum Day {
    MONDAY, TUESDAY, WEDNESDAY, THURSDAY, FRIDAY, SATURDAY, SUNDAY
}

再创建一个工人类

class Worker {
    String name;
    Set availableDays;

    public Worker(String name, Set availableDays) {
        this.name = name;
        this.availableDays = availableDays;
    }
    //省略getter和setter
}

创建几个工人对象

Worker[] workers = new Worker[]{
    new Worker("张三", EnumSet.of(Day.MONDAY, Day.TUESDAY, Day.WEDNESDAY, Day.FRIDAY)),
    new Worker("李四", EnumSet.of(Day.TUESDAY, Day.THURSDAY, Day.SATURDAY)),
    new Worker("王五", EnumSet.of(Day.TUESDAY, Day.THURSDAY))
};
  1. 一个人都不来的日子,即worker时间并集的补集

    private static void noBodyComeDay(Worker[] workers) {
        Set days = EnumSet.allOf(Day.class);
        for (Worker worker : workers) {
            days.removeAll(worker.getAvailableDays());
        }
        System.out.println(days);
    }
    
  2. 至少有一个人会来的日子,即worker时间的并集

    private static void atLeastOneBodyComeDay(Worker[] workers) {
        Set days = EnumSet.noneOf(Day.class);
        for(Worker w : workers){
            days.addAll(w.getAvailableDays());
        }
        System.out.println(days);
    }
    
  3. 所有人都会来的日子,即worker时间的交集

    private static void allManComeDay(Worker[] workers) {
        Set days = EnumSet.allOf(Day.class);
        for(Worker w : workers){
            days.retainAll(w.getAvailableDays());
        }
        System.out.println(days);
    }
    
  4. 哪些人周一和周二都会来

    private static void monAndTueComeBody(Worker[] workers) {
        Set availableWorkers = new HashSet<>();
        for(Worker w : workers){
            if(w.getAvailableDays().containsAll(EnumSet.of(Day.MONDAY, Day.TUESDAY))){
                availableWorkers.add(w);
            }
        }
        for(Worker w : availableWorkers){
            System.out.println(w.getName());
        }
    }
    
  5. 哪些天至少会有两个人来,先使用EnumMap统计每天的人数,然后找出至少有两个人的天(还没有完全弄懂)

    private static void atLeastTwoBodyComeDay(Worker[] workers) {
        Map countMap = new EnumMap<>(Day.class);
        for(Worker w : workers){
            for(Day d : w.getAvailableDays()){
                Integer count = countMap.get(d);
                countMap.put(d, count==null?1:count+1);
            }
        }
        Set days = EnumSet.noneOf(Day.class);//类型为Day不含任何元素
        for(Map.Entry entry : countMap.entrySet()){
            if(entry.getValue()>=2){
                days.add(entry.getKey());
            }
        }
        System.out.println(days);
    }
    

你可能感兴趣的:(EnumSet的使用)