【每日一题】1488. 避免洪水泛滥-2023.10.13

1488. 避免洪水泛滥

题目： 

你的国家有无数个湖泊，所有湖泊一开始都是空的。当第 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 天选择抽干的湖泊。

如果有多种可行解，请返回它们中的 任意一个 。如果没办法阻止洪水，请返回一个 空的数组 。

请注意，如果你选择抽干一个装满水的湖泊，它会变成一个空的湖泊。但如果你选择抽干一个空的湖泊，那么将无事发生。

示例 1：

输入：rains = [1,2,3,4]
输出：[-1,-1,-1,-1]
解释：第一天后，装满水的湖泊包括 [1]
第二天后，装满水的湖泊包括 [1,2]
第三天后，装满水的湖泊包括 [1,2,3]
第四天后，装满水的湖泊包括 [1,2,3,4]
没有哪一天你可以抽干任何湖泊的水，也没有湖泊会发生洪水。

示例 2：

输入：rains = [1,2,0,0,2,1]
输出：[-1,-1,2,1,-1,-1]
解释：第一天后，装满水的湖泊包括 [1]
第二天后，装满水的湖泊包括 [1,2]
第三天后，我们抽干湖泊 2 。所以剩下装满水的湖泊包括 [1]
第四天后，我们抽干湖泊 1 。所以暂时没有装满水的湖泊了。
第五天后，装满水的湖泊包括 [2]。
第六天后，装满水的湖泊包括 [1,2]。
可以看出，这个方案下不会有洪水发生。同时， [-1,-1,1,2,-1,-1] 也是另一个可行的没有洪水的方案。

示例 3：

输入：rains = [1,2,0,1,2]
输出：[]
解释：第二天后，装满水的湖泊包括 [1,2]。我们可以在第三天抽干一个湖泊的水。
但第三天后，湖泊 1 和 2 都会再次下雨，所以不管我们第三天抽干哪个湖泊的水，另一个湖泊都会发生洪水。

 解答：

这道题的核心是要找到每个湖泊两次下雨之间最早的晴天。 我们首先来分析一下什么时候是没法阻止洪水的【即返回空字符串】，那肯定是第 i 天下雨的湖泊 rains[i] 是满的，为什么是满的。是因为前面没有晴天可以把抽干【但凡有一天晴天都可以抽干这个湖泊】所以无法阻止洪水的情况是 当前下雨湖泊与其之前下雨的天数之间没有晴天。

因此只要两次下雨之间有晴天，这个湖泊肯定不会有洪水。那如果有多个晴天呢？我们要选择哪一个晴天来抽干这个水库呢？ 根据贪心的策略，我们应该选择最早的晴天进行抽干，把靠后的晴天留给更之后下雨的湖泊。 举个例子：

假设某个湖泊两次下雨的天数分别是 left 和 right ，那么我们的目标是找到 (left, right) 的首个晴天。【这里其实有个延迟处理的策略，即遇到晴天的时候我们不去考虑这个晴天要抽干哪个湖泊，而是等到后面某个湖泊第二次下雨时再来选择晴天进行处理】

因此我们可以使用一个 有序集合 记录遍历到的晴天，同时使用一个 哈希表 记录每个湖泊上一次下雨的天。 当我们遇到某个下雨天 j，在哈希表中能找到这一天的湖泊的上一个下雨天，即为 i，那我们就要在有序集合中找到首个大于 i 的晴天 k：

找到了，那么第 k 天就是用来抽干湖泊 rains[j]；同时从有序集合中移除这个晴天避免被重复找到；
没找到，无法阻止洪水；
由于我们依次遍历，晴天出现的天数是有序的。因此可以使用二分查找来快速找到集合中首个大于i的元素

还有一种可能就是晴天远比雨天，那么就会剩下一些晴天不是必须抽干湖泊的。我们默认多余晴天处理 1 号湖泊。

算法过程图解

代码：

class Solution {
    public int[] avoidFlood(int[] rains) {
        TreeSet<Integer> sunny = new TreeSet<>();   // 有序集合，记录可用的晴天
        Map<Integer, Integer> lakeLastRain = new HashMap<>();   // 记录每个湖泊上一次下雨的时间
        int n = rains.length;
        int[] ans = new int[n];
        Arrays.fill(ans, 1);    // 初始化ans所有值都为1，来使得剩余可用晴天默认抽干一号湖泊
        for(int i = 0; i < n; i++){
            if(rains[i] == 0){
                sunny.add(i);   // 将晴天的这一天加入有序集合
            }else{
                ans[i] = -1;    // 下雨天，这一天的ans肯定为-1
                if(lakeLastRain.containsKey(rains[i])){
                    // 如果这个湖泊之前也下过雨，我们就要两个下雨的区间内找到最早的不下雨天来抽空这个湖泊
                    Integer norain = sunny.ceiling(lakeLastRain.get(rains[i]));
                    if(norain == null)return new int[0];    // 如果没有晴天，肯定发洪无法阻止
                    ans[norain] = rains[i]; // 否则，找到的晴天抽空rains[i]湖泊
                    sunny.remove(norain);   // 这个晴天已经使用了，移除集合
                }
                lakeLastRain.put(rains[i], i);  // 记录rains[i]号湖泊这一天下过雨
            }
        }
        return ans;
    }
}

 结果：