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

相关推荐
嚣张农民7 小时前
推荐3个实用的760°全景框架
前端·vue.js·程序员
梓羽玩Python7 小时前
推荐一款用了5年的全能下载神器:Motrix!全平台支持,不限速下载网盘文件就靠它!
程序员·开源·github
梓羽玩Python8 小时前
这款一站式AI体验平台值得收藏起来!GPT-4o、GPT-4o Mini、Claude 3.5 Sonnet免费使用!
人工智能·程序员·设计
前端宝哥1 天前
10 个超赞的开发者工具,助你轻松提升效率
前端·程序员
XinZong1 天前
【VSCode插件推荐】想准时下班,你需要codemoss的帮助,分享AI写代码的愉快体验,附详细安装教程
前端·程序员
Goboy1 天前
0帧起步:3分钟打造个人博客,让技术成长与职业发展齐头并进
程序员·开源·操作系统
JaxNext1 天前
不选总统选配色,这一票投给 CSS logo
前端·css·程序员
程序员鱼皮2 天前
刚毕业,去做边缘业务,还有救吗?
计算机·程序员·互联网·求职·简历
WujieLi3 天前
独立开发沉思录周刊:vol18.AI 正在成为无处不在的基础设施
程序员·设计·创业
_祝你今天愉快3 天前
重学Android:从位运算到二进制表示(零)
算法·程序员