贪心算法（集合覆盖）

1、基本介绍 

2、应用实例（集合覆盖）

package algorithm;

import java.util.*;

public class Greedy {
    public static void main(String[] args) {
        String[] areas = {"上海","天津","北京","广州","深圳","成都","杭州","大连"};
        List<String> allAreas = arrayToList(areas);
        Map<String, Set> map = createBroadCast();
        printMap(map);
        List<String> broadcasts = greedy(allAreas, map);
        System.out.println("经贪心算法选中的广播电台：" + broadcasts);
    }

    //  贪心算法
    private static List<String> greedy(List<String> allAreas, Map<String, Set> map) {
        String maxKey;                                  // 指向未覆盖的最大地区
        List<String> selected = new ArrayList<>();      // 存放选中的电台（存放结果的）
        Set<String> tempSet = new HashSet<>();          // 存放的是maxKey指向的最大覆盖地区和allArea中未覆盖地区的交集
        while (allAreas.size() > 0 ) {                  // allAreas保存的是所有未覆盖地区（为空时，表示均已覆盖）
            maxKey = null;                              // maxKey每次使用前都应先指向null
            for (String key : map.keySet()) {           // 遍历map中每一个key（即电台）
                tempSet.clear();                        // 每次使用完都要清空，因为只做暂存的作用
                tempSet = map.get(key);
                tempSet.retainAll(allAreas);            // 求交集，获取当前未覆盖的地区，保存到tempSet中
                if (tempSet.size() > 0 &&               // 如果tempSet>0表明当前key有未覆盖地区，再将其与（maxKey指向的）上一个电台未覆盖地区进行比较，随即找出最大覆盖地区（贪心算法的精髓）
                        (maxKey == null || tempSet.size() > map.get(maxKey).size())) {
                    maxKey = key;
                }
                if (maxKey != null && tempSet.size() > 0) {                   // 不为null时，表示已指向最大未覆盖地区
                    selected.add(maxKey);               // 将当前电台添加到结果集合（selected）中
                    allAreas.removeAll(map.get(maxKey));// 把已选中的未覆盖地区从集合（allAreas）中删除
                }
            }
        }
        return selected;
    }


    //  建立广播电台和覆盖地区的映射
    private static Map<String, Set> createBroadCast() {
        Map<String, Set> map = new HashMap<>();
        Set<String> set1 = new HashSet<>();
        Set<String> set2 = new HashSet<>();
        Set<String> set3 = new HashSet<>();
        Set<String> set4 = new HashSet<>();
        Set<String> set5 = new HashSet<>();
        set1.add("北京");
        set1.add("上海");
        set1.add("天津");
        set2.add("广州");
        set2.add("北京");
        set2.add("深圳");
        set3.add("成都");
        set3.add("上海");
        set3.add("杭州");
        set4.add("上海");
        set4.add("天津");
        set5.add("杭州");
        set5.add("大连");
        map.put("K1", set1);
        map.put("K2", set2);
        map.put("K3", set3);
        map.put("K4", set4);
        map.put("K5", set5);
        return map;
    }

    //  将数组转为集合
    private static List arrayToList(String[] areas) {
        List allAreas = new ArrayList<String>();
        for (String temp : areas) {
            allAreas.add(temp);
        }
        return allAreas;
    }

    //  打印map集合
    public static void printMap(Map map) {
        for (Object entry : map.entrySet()) {
            System.out.println(entry);
        }
    }
}