leetcode_378 Kth Smallest Element in a Sorted Matrix

最新推荐文章于 2022-10-28 23:16:46 发布

Dream_yz

最新推荐文章于 2022-10-28 23:16:46 发布

阅读量739

点赞数

分类专栏： leetcode编程 leetcode刷题文章标签： leetcode 378 C++实现有序矩阵第k个数

本文链接：https://blog.csdn.net/yzhang6_10/article/details/52541809

版权

leetcode编程同时被 2 个专栏收录

92 篇文章 1 订阅

订阅专栏

leetcode刷题

92 篇文章 2 订阅

订阅专栏

题目分析

给定一个n*n的矩阵，其中每一行每一列都按照递增排序，寻找矩阵中的第k小元素。注意，是要寻找排好序的第k小元素，而不是第k个不重复元素。
解题思路

1）利用堆来实现。直接维护一个大小为k的堆，全部读入一边，这样平均和最好时间情况都为O(n*logk)。
2）利用二分查找实现。
C++实现

struct Node
{
    int val;
    int i; 
    int j;
    Node(int i, int j, int val) : i(i), j(j), val(val) { }
    bool operator < (const Node& x) const
    {
        return val > x.val;
    }
};
// 利用堆实现
int kthSmallest1(vector< vector<int> >& matrix, int k)
{
    priority_queue<Node> que;
    int n = matrix.size();
    que.push(Node(0, 0, matrix[0][0]));
    while (--k)
    {
        Node x = que.top();
        que.pop();
        if (x.i == 0 && x.j + 1 < n)
            que.push(Node(x.i, x.j + 1, matrix[x.i][x.j + 1]));
        if (x.i + 1 < n)
            que.push(Node(x.i + 1, x.j, matrix[x.i + 1][x.j]));
    }
    return que.top().val;
}
// 利用二分查找实现
int kthSmallest(vector< vector<int> >& matrix, int k)
{
    int n = matrix.size();
    int L = matrix[0][0];
    int R = matrix[n - 1][n - 2];
    while (L < R)
    {
        int mid = L + ((R - L) >> 1);
        int temp = 0;
        for (int i = 0; i < n; i++)
            temp += upper_bound(matrix[i].begin(), matrix[i].end(), mid) - matrix[i].begin();
        if (temp < k)
            L = mid + 1;
        else
            R = mid + 1;
    }
    return L;
}