题目
题解
- 默认对所有天数填写默认值,即都抽 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多平台发布