力扣1488.避免洪水泛滥

力扣1488.避免洪水泛滥

  • 贪心 + 二分

    • 将所有晴天存入集合
    • 用哈希表存每次池子上一次下雨的日期
    • 当下雨并且池子满了时,二分找到上一次下雨之后最近的晴天
cpp 复制代码
  class Solution {
      unordered_map<int,int> mp;
  public:
      vector<int> avoidFlood(vector<int>& rains) {
          int n = rains.size();
          vector<int> ans(n,1);
          set<int> zero;
          for(int i=0;i<n;i++)
          {
              int r = rains[i];
              //晴天存入集合
              if(r == 0)
              {
                  zero.insert(i);
                  continue;
              }
              //下雨并且满了
              if(mp.count(r) != 0)
              {
                  //mp[r]为上一次下雨的日期 lowerbound找>mp[r]的第一个晴天日期
                  auto it = zero.lower_bound(mp[r]);
                  if(it == zero.end()) return {};
                  ans[*it] = r;
                  zero.erase(it);
              }
              //本次下雨池子和日期
              mp[r] = i;
              ans[i] = -1;
          }
          return ans;
      }
  };
相关推荐
随风75643 分钟前
有多少小于当前数字的数字 力扣1365
数据结构·算法·leetcode
今天吃什么了1 小时前
矩阵幂(矩阵k次幂)
人工智能·算法·矩阵
我叫王棱夹1 小时前
第五章-动态规划
算法·动态规划
dot to one2 小时前
深入刨析C++ stl::stack 、stl::queue 及 priority_queue
c语言·开发语言·数据结构·c++·算法·visual studio
程序趣谈5 小时前
算法随笔_74: 不同路径_1
数据结构·python·算法
修修修也6 小时前
算法手记3
数据结构·学习·算法·刷题
吧啦吧啦吡叭卜7 小时前
【打卡d5】快速排序 归并排序
java·算法·排序算法
L_cl7 小时前
【Python 数据结构 15.哈希表】
数据结构·算法·散列表
肖筱小瀟8 小时前
2025-3-13 leetcode刷题情况(贪心算法--区间问题)
算法·leetcode·贪心算法