数据结构与算法学习之贪心算法

最新推荐文章于 2024-10-15 10:20:47 发布

爱吃早饭的小王

最新推荐文章于 2024-10-15 10:20:47 发布

阅读量59

点赞数

文章标签：学习贪心算法 java

本文链接：https://blog.csdn.net/m0_55895641/article/details/126380138

版权

package com.atguigu.greedy;

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

/**
 * @author 
 * @create 2022-08-17-9:14
 */
public class GreedyAlgorithm {
    public static void main(String[] args) {
        //创建广播电台，放入Map中
        HashMap<String, HashSet<String>> broadcasts = new HashMap<>();
        //将各个电台放入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("大连");

        //selects存放选择的电台
        ArrayList<String> selects = new ArrayList<>();

        //tempSet用来存放遍历的电台覆盖的城市，及后续与未覆盖集合求交集
        HashSet<String> tempSet = new HashSet<>();
        //定义maxKey，保存遍历过程中未覆盖城市最多的电台
        //如果maxKey不为null则加入到selects中
        String maxKey = null;

        //遍历所有电台
        while (allAreas.size() != 0) {
            //每进行一次循环需要将maxKey置空
            maxKey = null;


            //每一次for循环遍历一遍所有电台，找其中最优的
            for (String key : broadcasts.keySet()) {
                //每进行一次for循环，需要将tempSet清空
                tempSet.clear();

                //找到每个电台对应的覆盖城市
                HashSet<String> areas = broadcasts.get(key);
                //将这些城市放入临时set中
                tempSet.addAll(areas);

                //临时集合与剩余未覆盖城市求交集，得到还未覆盖城市放入到tempSet中
                tempSet.retainAll(allAreas);

                //改变maxKey   tempSet.size() > broadcasts.get(maxKey).size()体现出贪心算法的特点，每次都选择最优
                if (tempSet.size() > 0 && (maxKey == null || tempSet.size() > broadcasts.get(maxKey).size())) {
                    maxKey = key;
                }

            }
            //如果maxKey不为空，则加入到selects中
            if (maxKey != null) {
                selects.add(maxKey);
                //将maxKey选中后，在allAreas中将maxKey代表的城市去掉
                allAreas.removeAll(broadcasts.get(maxKey));
            }
        }

        System.out.println("得到的选择结果"+selects);//[k1 k2 k3 k5]
    }
}