1488. 避免洪水泛滥

题目

题解

  • 默认对所有天数填写默认值,即都抽 1 号湖泊
  • 用 Map 存储湖泊编号和下雨天的对应关系
  • 把所有的空闲日子(不下雨的天数)存起来,注意要使用 TreeSet,因为后面需要进行比较
  • 遍历数组
    • 不下雨:将 i 天存储到 TreeSet 中,等后面再选择这一天抽哪个湖泊
    • 下雨:如果下雨的湖泊已经有水,则判断有水后至今期间有没有空闲日子
      • 有空闲:设置那天为抽水,并移除该空闲日子,使用 TreeSet.ceiling 方法进行比较
      • 无空闲:直接返回空数字
java 复制代码
class Solution {
    public int[] avoidFlood(int[] rains) {
        int[] res = new int[rains.length];
        Arrays.fill(res, 1);
        TreeSet<Integer> freeDay = new TreeSet<>();
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < rains.length; i++) {
            int rain = rains[i];
            if (rain == 0) {
                // 空闲日子
                freeDay.add(i);
            } else {
                // 下雨
                res[i] = -1;
                if (map.containsKey(rain)) {
                    // 已经有水,则抽掉
                    Integer tp = freeDay.ceiling(map.get(rain));
                    if (tp == null) {
                        return new int[]{};
                    } else {
                        // 用之前存储的空闲天数抽掉 rain 湖泊的水
                        res[tp] = rain;
                        freeDay.remove(tp);
                    }
                }
                map.put(rain, i);
            }
        }
        return res;
    }
}

本文由mdnice多平台发布

相关推荐
用户35391699362154 小时前
DeepSeek回答过于笼统,提示词如何优化
程序员
摆烂工程师4 小时前
ChatGPT免费用户可以使用Deep Research啦!并且o3、o4-mini的可使用次数翻倍!
前端·后端·程序员
程序员辣条4 小时前
深度测评 RAG 应用评估框架:指标最全面的 RAGas
人工智能·程序员
程序员辣条4 小时前
为什么需要提示词工程?什么是提示词工程(prompt engineering)?为什么需要提示词工程?收藏我这一篇就够了!
人工智能·程序员·产品经理
爱喝奶茶的企鹅16 小时前
Ethan独立开发产品日报 | 2025-04-24
人工智能·程序员·开源
Captaincc18 小时前
告别~!烦人的 homebrew 强制更新
程序员
SimonKing1 天前
惊!未实现Serializable竟让第三方接口回调全军覆没
前端·程序员·架构
哔哩哔哩技术1 天前
世界知识产权日 | 2025年度哔哩哔哩技术专利评选结果出炉!
程序员
Lx3521 天前
跨国团队协作的十二时辰:时区、文化、代码的三重奏
程序员
TF男孩2 天前
中年程序员,看不惯大公司,去小公司了
程序员