力扣632.最小区间

力扣632.最小区间

  • 贪心 + 最小堆

    • 用一个小根堆维护K个数
    • 其中最小的算完结果后弹出,再补一个进去
cpp 复制代码
  class Solution {
  public:
      vector<int> smallestRange(vector<vector<int>>& nums) {
          int l=0,r=INT_MAX;
          int n = nums.size();
          //记录下一个位置的下标
          vector<int> next(n);
          //按照数值排序
          auto cmp = [&](const int &u,const int &v)
          {
              return nums[u][next[u]] > nums[v][next[v]];
          };
          //
          priority_queue<int,vector<int>,decltype(cmp)> pq(cmp);
          //当前的上下限
          int minval = 0,maxval = INT_MIN;
          //把所有列表放进去
          for(int i=0;i<n;i++)
          {
              pq.emplace(i);
              maxval = max(maxval,nums[i][0]);
          }
          while(true)
          {
              //最小数的下标
              int row = pq.top();
              pq.pop();
              minval = nums[row][next[row]];
              //更优
              if(maxval - minval < r - l)
                  l = minval , r = maxval;
              //当前列表全部遍历过了
              if(next[row] == nums[row].size() - 1)
                  break;
              //next往后移动一位
              next[row] ++;
              //更新当前k个数中最大值
              maxval = max(maxval,nums[row][next[row]]);
              pq.emplace(row);
          }
          return {l,r};
      }
  };
相关推荐
再卷也是菜4 分钟前
C++篇(14)二叉树进阶算法题
c++·算法
小邓儿◑.◑8 分钟前
贪心算法 | 每周8题(三)
算法·贪心算法
2401_8414956411 分钟前
【数据结构】最长的最短路径的求解
java·数据结构·c++·python·算法·最短路径·图搜索
小龙报14 分钟前
《算法每日一题(1)--- 连续因子》
c语言·开发语言·c++·windows·git·算法·visual studio
夜晚中的人海40 分钟前
【C++】滑动窗口算法习题
开发语言·c++·算法
violet-lz1 小时前
数据结构四大简单排序算法详解:直接插入排序、选择排序、基数排序和冒泡排序
数据结构·算法·排序算法
·白小白1 小时前
力扣(LeetCode) ——118.杨辉三角(C++)
c++·算法·leetcode
CoovallyAIHub2 小时前
超越“识别”:下一代机器视觉如何破解具身智能落地难题?
深度学习·算法·计算机视觉
仰泳的熊猫2 小时前
LeetCode:207. 课程表
数据结构·c++·算法·leetcode