贪心算法
算法简介
贪心算法又称贪婪算法,它的核心思想就是不管要做什么选择,或是得到什么结果,都要最好的最优的,跟名字一样比较贪婪,但是最后得到的结果不一定是最好的。可能是最好的,或是无限接近于最好的结果。
电台问题
如果有如图这些广播台,以及广播台信号可以覆盖的地区。 如何选择最少的广播台,让所有的地区都可以接收到信号
问题分析
首先我们可以把所有需要覆盖到的所有地区都存入一个集合,然后遍历这些电台,找到可以覆盖最多的一个,然后拿出,并且从集合中去掉已经覆盖的城市。依次类推,直至存地区的集合为空,问题解决。
代码实现
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);
}
}
总结
贪心算法的思想就是,不管哪一步,都要取出最优值进行运算,直至结束。我们要思考的就是如何每一步都取出最优值。
秃头萌新一枚 多多关照