码迷,mamicode.com
首页 > 其他好文 > 详细

kth-smallest-element-in-a-sorted-matrix

时间:2016-08-02 14:59:14      阅读:116      评论:0      收藏:0      [点我收藏+]

标签:

//有很多讨论,比如
// https://discuss.leetcode.com/topic/52865/my-solution-using-binary-search-in-c
// https://discuss.leetcode.com/topic/52912/binary-search-heap-and-sorting-comparison-with-concise-code-and-1-liners-python-72-ms

class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        // 注意以下算法是直接按照整数范围>>1来进行二分查找的
        int row = matrix.size(), col = matrix[0].size();
        int small = matrix[0][0], big = matrix[row-1][col-1];
        int mid;
        while (small < big) {
            mid = (small + big) >> 1;
            int len = 0;
            for (int i=0; i<row; ++i) {
                len += upper_bound(matrix[i].begin(), matrix[i].end(), mid) - matrix[i].begin();
            }
            if (len < k) {
                small = mid + 1;
            }
            else {
                big = mid;
            }
        }
        return small;
    }
};

 

kth-smallest-element-in-a-sorted-matrix

标签:

原文地址:http://www.cnblogs.com/charlesblc/p/5729230.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!