算法 - 贪心算法(集合覆盖问题求解)

1.穷举法

2.贪心算法

遍历集合的key，用当前key的value和allAreas去取交集），然后和（maxKey和allAreas的交集）比较大小，如果当前key的结合size大就把当前key赋给maxkey，循环5次后退出循环，并把maxkey加入到selects，且把maxkey覆盖的地区从allAreas中移除。直到allAreas集合为空退出while循环，得到的select就是所需要的。

package Algorithm.greedy;

import java.util.*;

public class GreedyAlogrithm {
    public static void main(String[] args) {
        //创建广播电台，放入map
        HashMap<String, HashSet<String>> broadcasts = new HashMap<>();
        //将各个电台放入
        HashSet<String> set1 = new HashSet<>();
        set1.add("北京");
        set1.add("上海");
        set1.add("天津");
        HashSet<String> set2 = new HashSet<>();
        set2.add("广州");
        set2.add("北京");
        set2.add("深圳");
        HashSet<String> set3 = new HashSet<>();
        set3.add("成都");
        set3.add("上海");
        set3.add("杭州");
        HashSet<String> set4 = new HashSet<>();
        set4.add("上海");
        set4.add("天津");
        HashSet<String> set5 = new HashSet<>();
        set5.add("杭州");
        set5.add("大连");

        //加入到map
        broadcasts.put("k1",set1);
        broadcasts.put("k2",set2);
        broadcasts.put("k3",set3);
        broadcasts.put("k4",set4);
        broadcasts.put("k5",set5);

        //创建一个存放所有城市的集合，就遍历集合就行
        HashSet<String> allAreas = new HashSet<>();
        for(Map.Entry<String, HashSet<String>> entry: broadcasts.entrySet())
        {
            HashSet<String> value = entry.getValue();
            Iterator<String> iterator1 = value.iterator();
            while (iterator1.hasNext()){
                allAreas.add(iterator1.next());
            }
        }

        //System.out.println(allAreas);

        //创建一个ArrayList存放选择后的 电台的集合
        ArrayList<String> selects = new ArrayList<>();
        //保存每次 当前覆盖的地区和未覆盖到的地区的交集，用retainAll()方法
        HashSet<String> tempSet = new HashSet<>();
        HashSet<String> value = new HashSet<>();
        //创建一个指针maxKey,存放交集最大的key
        String maxKey = null;

        while (allAreas.size() != 0){ //allAreas 不为空就表示没覆盖到所有地区
            maxKey = null;

            for (String key : broadcasts.keySet()) {
                tempSet.clear();
                value.clear();
                //存放Key覆盖的地区
                value.addAll(broadcasts.get(key));
                //tempSet.addAll(value);
                //求当前key覆盖的地区和allAreas的交集
                value.retainAll(allAreas);
                //下面两步是取出maxKey覆盖地区和allAreas的交集
                if (maxKey != null) {
                    tempSet.addAll(broadcasts.get(maxKey));
                    tempSet.retainAll(allAreas);
                }

                if (value.size() > 0 && (maxKey == null || (value.size() > tempSet.size()))){
                    maxKey = key;
                }
            }
            if (maxKey != null) {
                selects.add(maxKey);
                    allAreas.removeAll(broadcasts.get(maxKey));
            }
        }

        System.out.println(selects);
    }
}