EnumSet的使用

应用场景一

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

实现

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

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

再创建一个工人类

class Worker {
    String name;
    Set<Day> availableDays;

    public Worker(String name, Set<Day> 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<Day> 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<Day> 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<Day> 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<Worker> 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<Day, Integer> 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<Day> days = EnumSet.noneOf(Day.class);//类型为Day不含任何元素
       for(Map.Entry<Day, Integer> entry : countMap.entrySet()){
           if(entry.getValue()>=2){
               days.add(entry.getKey());
           }
       }
       System.out.println(days);
   }