当前位置：首页 > news >正文

Leetcode—1488.避免洪水泛滥【中等】

news 2025/7/11 21:22:05

2023每日刷题（十四）

Leetcode—1488.避免洪水泛滥

算法思想

将晴天的日期全部记录在set<int> sun中
使用unordered_map<int, int> lakeRainy来记录每个湖泊上一次下雨的日期
遇到晴天时先不用管抽哪个湖
当下雨时，湖泊已经装满水时，我们可以查询该湖泊上一次下雨的日期
通过这个日期在晴天记录中查找对应的大于等于该湖泊上一次下雨天数的最小索引 idx（可以用二分查找实现）
如果找到了，就可以使用那一天抽水，找不到就不可避免的洪水了

实现代码

class Solution {
public:
    vector<int> avoidFlood(vector<int>& rains) {
        int n = rains.size();
        vector<int> ans(n, 1);
        set<int> sun;
        unordered_map<int, int> lakeRainy;

        int i;
        for(i = 0; i < n; i++) {
            // 记录晴天的日期
            if(rains[i] == 0) {
                sun.insert(i);
                continue;
            }
            if(lakeRainy.count(rains[i]) != 0) {
                auto t = sun.lower_bound(lakeRainy[rains[i]]);
                if(t == sun.end()) {
                    return {};
                }
                ans[*t] = rains[i];
                sun.erase(t);
            }
            ans[i] = -1;
            lakeRainy[rains[i]] = i;
        }
        return ans;
    }
};