贪心算法详解

java代码

package greedy;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;

public class GreedyAlgorithm {

    public static void main(String[] args) {
        //1 创建广播电台，放入到Map
        HashMap<String, HashSet<String>> broadcasts = new HashMap<>();

        //2 将各个电台放入到broadcasts
        HashSet<String> hashset1 = new HashSet<>();
        hashset1.add("北京");
        hashset1.add("上海");
        hashset1.add("天津");

        HashSet<String> hashset2 = new HashSet<>();
        hashset2.add("广州");
        hashset2.add("北京");
        hashset2.add("深圳");

        HashSet<String> hashset3 = new HashSet<>();
        hashset3.add("成都");
        hashset3.add("上海");
        hashset3.add("杭州");

        HashSet<String> hashset4 = new HashSet<>();
        hashset4.add("上海");
        hashset4.add("天津");

        HashSet<String> hashset5 = new HashSet<>();
        hashset5.add("杭州");
        hashset5.add("大连");

        //加入到map
        broadcasts.put("k1", hashset1);
        broadcasts.put("k2", hashset2);
        broadcasts.put("k3", hashset3);
        broadcasts.put("k4", hashset4);
        broadcasts.put("k5", hashset5);

        //allAreas 存放所有的地区
        HashSet<String> allAreas = new HashSet<>();
        allAreas.add("北京");
        allAreas.add("上海");
        allAreas.add("天津");
        allAreas.add("广州");
        allAreas.add("深圳");
        allAreas.add("成都");
        allAreas.add("杭州");
        allAreas.add("大连");

        //1 创建一个存放结果的集合
        HashSet<String> selects = new HashSet<>();
        //2 创建一个临时集合，存放每个电台与未匹配城市的交集
        HashSet<String> tempSet = new HashSet<>();
        //3 定义存储能匹配到最大未匹配城市的key,记为maxKey
        String maxKey;
        // 定义匹配的初始大小
        int count;
        //4 循环遍历城市集合，直到为空
        while (allAreas.size() != 0) {
            count = 0;
            maxKey = null;
            //4.1 循环
            for (String key : broadcasts.keySet()) {
                tempSet.clear();
                //4.1.1 将其加入tempSet集合
                tempSet.addAll(broadcasts.get(key));
                //4.1.2 取当前key和当前所有未匹配城市的交集
                tempSet.retainAll(allAreas);
                //4.1.3 比较当前交集与此前最大值的大小
                if (tempSet.size() > count) {
                    maxKey = key;
                    count = tempSet.size();
                }
            }
            //
            if (maxKey != null) {
                selects.add(maxKey);
                allAreas.removeAll(broadcasts.get(maxKey));
            }
        }
        System.out.println(selects);
    }
}