1、基本介绍
2、应用实例(集合覆盖)
package algorithm;
import java.util.*;
public class Greedy {
public static void main(String[] args) {
String[] areas = {"上海","天津","北京","广州","深圳","成都","杭州","大连"};
List<String> allAreas = arrayToList(areas);
Map<String, Set> map = createBroadCast();
printMap(map);
List<String> broadcasts = greedy(allAreas, map);
System.out.println("经贪心算法选中的广播电台:" + broadcasts);
}
// 贪心算法
private static List<String> greedy(List<String> allAreas, Map<String, Set> map) {
String maxKey; // 指向未覆盖的最大地区
List<String> selected = new ArrayList<>(); // 存放选中的电台(存放结果的)
Set<String> tempSet = new HashSet<>(); // 存放的是maxKey指向的最大覆盖地区和allArea中未覆盖地区的交集
while (allAreas.size() > 0 ) { // allAreas保存的是所有未覆盖地区(为空时,表示均已覆盖)
maxKey = null; // maxKey每次使用前都应先指向null
for (String key : map.keySet()) { // 遍历map中每一个key(即电台)
tempSet.clear(); // 每次使用完都要清空,因为只做暂存的作用
tempSet = map.get(key);
tempSet.retainAll(allAreas); // 求交集,获取当前未覆盖的地区,保存到tempSet中
if (tempSet.size() > 0 && // 如果tempSet>0表明当前key有未覆盖地区,再将其与(maxKey指向的)上一个电台未覆盖地区进行比较,随即找出最大覆盖地区(贪心算法的精髓)
(maxKey == null || tempSet.size() > map.get(maxKey).size())) {
maxKey = key;
}
if (maxKey != null && tempSet.size() > 0) { // 不为null时,表示已指向最大未覆盖地区
selected.add(maxKey); // 将当前电台添加到结果集合(selected)中
allAreas.removeAll(map.get(maxKey));// 把已选中的未覆盖地区从集合(allAreas)中删除
}
}
}
return selected;
}
// 建立广播电台和覆盖地区的映射
private static Map<String, Set> createBroadCast() {
Map<String, Set> map = new HashMap<>();
Set<String> set1 = new HashSet<>();
Set<String> set2 = new HashSet<>();
Set<String> set3 = new HashSet<>();
Set<String> set4 = new HashSet<>();
Set<String> set5 = new HashSet<>();
set1.add("北京");
set1.add("上海");
set1.add("天津");
set2.add("广州");
set2.add("北京");
set2.add("深圳");
set3.add("成都");
set3.add("上海");
set3.add("杭州");
set4.add("上海");
set4.add("天津");
set5.add("杭州");
set5.add("大连");
map.put("K1", set1);
map.put("K2", set2);
map.put("K3", set3);
map.put("K4", set4);
map.put("K5", set5);
return map;
}
// 将数组转为集合
private static List arrayToList(String[] areas) {
List allAreas = new ArrayList<String>();
for (String temp : areas) {
allAreas.add(temp);
}
return allAreas;
}
// 打印map集合
public static void printMap(Map map) {
for (Object entry : map.entrySet()) {
System.out.println(entry);
}
}
}