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