1488. 避免洪水泛滥-526互联

题目

题解

默认对所有天数填写默认值，即都抽 1 号湖泊
用 Map 存储湖泊编号和下雨天的对应关系
把所有的空闲日子（不下雨的天数）存起来，注意要使用 TreeSet，因为后面需要进行比较
遍历数组
- 不下雨：将 i 天存储到 TreeSet 中，等后面再选择这一天抽哪个湖泊
- 下雨：如果下雨的湖泊已经有水，则判断有水后至今期间有没有空闲日子
  - 有空闲：设置那天为抽水，并移除该空闲日子，使用 TreeSet.ceiling 方法进行比较
  - 无空闲：直接返回空数字

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