你的国家有无数个湖泊,所有湖泊一开始都是空的。当第 n 个湖泊下雨前是空的,那么它就会装满水。如果第 n 个湖泊下雨前是 满的 ,这个湖泊会发生 洪水 。你的目标是避免任意一个湖泊发生洪水。
给你一个整数数组 rains ,其中:
rains[i] > 0 表示第 i 天时,第 rains[i] 个湖泊会下雨。
rains[i] == 0 表示第 i 天没有湖泊会下雨,你可以选择 一个 湖泊并 抽干 这个湖泊的水。
请返回一个数组 ans ,满足:
ans.length == rains.length
如果 rains[i] > 0 ,那么ans[i] == -1 。
如果 rains[i] == 0 ,ans[i] 是你第 i 天选择抽干的湖泊。
如果有多种可行解,请返回它们中的 任意一个 。如果没办法阻止洪水,请返回一个 空的数组 。
请注意,如果你选择抽干一个装满水的湖泊,它会变成一个空的湖泊。但如果你选择抽干一个空的湖泊,那么将无事发生。
思路:
洪水定义为湖泊里面有水,但仍继续下雨。rains[i]=0时,第i天可以抽干任意湖泊中的水。
使用贪心算法,当第i天的第j个湖泊发生洪水时,看看第j个湖泊刚有水后面的日子,有没有可以抽水的天,如果有,则直接在那天抽水,没有直接返回空集。
使用HashMap存储湖泊和降水的日期。使用TreeSet存储可以进行抽干操作的日期。如果第j的湖泊降雨时候,看看HashMap中是否有该湖泊,如果有,取day=treeset.higher(hashMap.get(lake)),有就更新ans[day]为lake,没有就返回空集。贪心算法需要一个清晰的思路。
class Solution {public int[] avoidFlood(int[] rains) {int n = rains.length;// 保存不下雨的天数TreeSet<Integer> dryDay = new TreeSet();// 保存下雨的湖泊与天数Map<Integer, Integer> map = new HashMap();int[] ans = new int[n];Arrays.fill(ans, 1);for(int i=0; i<rains.length; i++) {int lake = rains[i];if(lake == 0) {dryDay.add(i);} else {ans[i] = -1;if(map.containsKey(lake)) {Integer higher = dryDay.higher(map.get(lake));if(higher == null) {return new int[]{};}dryDay.remove(higher);ans[higher] = lake;}// 更新下雨的湖泊与日期map.put(lake, i);}} return ans;}
}