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多平台发布

相关推荐
congregalis17 小时前
跳出框架:一步步实现简易Deep Search Agent
人工智能·程序员·源码
仙灵灵1 天前
前端的同学看过来,今天讲讲jwt登录
前端·后端·程序员
陈随易1 天前
VSCode v1.99发布,王者归来,Agent和MCP正式推出
前端·后端·程序员
amagi6002 天前
关于我在大学学编程之后一些学习思维的杂谈
程序员
袁煦丞2 天前
【亲测】1.5万搞定DeepSeek满血版!本地部署避坑指南+内网穿透黑科技揭秘
人工智能·程序员·远程工作
我要改名叫嘟嘟2 天前
读过的书会全部忘掉么?不会的,总有些会在生命中留下印记
程序员
马可奥勒留2 天前
我的管理日记(2)——招聘
程序员
小华同学ai3 天前
1K star!这个开源项目让短信集成简单到离谱,开发效率直接翻倍!
后端·程序员·github
程序员鱼皮3 天前
感觉程序员要被 AI 淘汰了?学什么才有机会?
计算机·ai·程序员·互联网·编程经验