贪心算法_电台覆盖问题

贪心算法

算法简介

贪心算法又称贪婪算法，它的核心思想就是不管要做什么选择，或是得到什么结果，都要最好的最优的，跟名字一样比较贪婪，但是最后得到的结果不一定是最好的。可能是最好的，或是无限接近于最好的结果。

电台问题

如果有如图这些广播台，以及广播台信号可以覆盖的地区。 如何选择最少的广播台，让所有的地区都可以接收到信号

问题分析

首先我们可以把所有需要覆盖到的所有地区都存入一个集合，然后遍历这些电台，找到可以覆盖最多的一个，然后拿出，并且从集合中去掉已经覆盖的城市。依次类推，直至存地区的集合为空，问题解决。

代码实现

public class Greedy {
public static void main(String[] args){
  // 先创建广播电台 放入map进行管理
  HashMap<String, HashSet<String>> broadcast = 
  new HashMap<>();

  // 将各个电台放入到broadcast
  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("大连");

  broadcast.put("k1",hashSet1);
  broadcast.put("k2",hashSet2);
  broadcast.put("k3",hashSet3);
  broadcast.put("k4",hashSet4);
  broadcast.put("k5",hashSet5);

  // 存放所有的地区
  HashSet<String> allAreas = new HashSet<>();


  // 遍历所有城市 存放到allAreas
  for (Map.Entry<String, HashSet<String>> 
  entry : broadcast.entrySet()) {
  
      Iterator<String> it = 
      entry.getValue().iterator();
      
      while (it.hasNext()) {
          String next = it.next();
          allAreas.add(next);
      }
  }
// 创建Arraylist存放选择的电台集合
ArrayList<String> select = new ArrayList<>();

// 定义一个临时集合 在遍历过程中 存放遍历过程中的电台覆盖的地区和当前还没覆盖地区的交集
HashSet<String> tempSet = new HashSet<>();

// 定义maxKey保存在一次遍历过程中能够覆盖最多未覆盖地区地区的电台对应的key
String maxKey = null;
// 如果key不为空 则假如select
// 如果allAreas不为0 则还没有覆盖到所有地区
while (allAreas.size() != 0){

maxKey = null;

for (String key : broadcast.keySet()) {

  tempSet.clear();

  HashSet<String> areas = broadcast.get(key);
  tempSet.addAll(areas);
  // 求出tempSet和allAreas的交集 然后赋给tempSet
  tempSet.retainAll(allAreas);


if (tempSet.size() > 0 && (maxKey == null ||
tempSet.size() > broadcast.get(maxKey).size())){
   maxKey = key;
}
}
if (maxKey != null){
    select.add(maxKey);

allAreas.removeAll(broadcast.get(maxKey));

        }
}

  System.out.println("结果"+select);

}
}

总结

贪心算法的思想就是，不管哪一步，都要取出最优值进行运算，直至结束。我们要思考的就是如何每一步都取出最优值。

秃头萌新一枚 多多关照