力扣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};
      }
  };
相关推荐
蒋星熠12 小时前
Flutter跨平台工程实践与原理透视:从渲染引擎到高质产物
开发语言·python·算法·flutter·设计模式·性能优化·硬件工程
小欣加油13 小时前
leetcode 面试题01.02判定是否互为字符重排
数据结构·c++·算法·leetcode·职场和发展
3Cloudream13 小时前
LeetCode 003. 无重复字符的最长子串 - 滑动窗口与哈希表详解
算法·leetcode·字符串·双指针·滑动窗口·哈希表·中等
王璐WL13 小时前
【c++】c++第一课:命名空间
数据结构·c++·算法
空白到白14 小时前
机器学习-聚类
人工智能·算法·机器学习·聚类
索迪迈科技14 小时前
java后端工程师进修ing(研一版 || day40)
java·开发语言·学习·算法
zzzsde14 小时前
【数据结构】队列
数据结构·算法
芒克芒克14 小时前
LeetCode 面试经典 150 题:删除有序数组中的重复项(双指针思想解法详解)
算法
青 .14 小时前
数据结构---二叉搜索树的实现
c语言·网络·数据结构·算法·链表