leetcode 378/668 第k小的元素（二分法）

题目描述：
378：
给定一个 n x n 矩阵，其中每行和每列元素均按升序排序，找到矩阵中第 k 小的元素。
请注意，它是排序后的第 k 小元素，而不是第 k 个不同的元素。
668：
给定一个 n x n 矩阵，其中每行和每列元素均按升序排序，找到矩阵中第 k 小的元素。
请注意，它是排序后的第 k 小元素，而不是第 k 个不同的元素。

思路：
值域二分法；
然后计算不大于mid的个数
按行计算

代码如下：
378：

class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        int m=matrix.size();
        int n=matrix[0].size();
        if(k==1) return matrix[0][0];
        if(k==m*n) return matrix[m-1][n-1];
        int left=matrix[0][0],right=matrix[m-1][n-1],mid;
        while(left<right){
            mid=left+(right-left)/2;
            int temp=fun(matrix,mid);
            if(temp<k){
                left=mid+1;
            }
            else{
                right=mid;
            }
        }
        return left;
    }
    int fun(vector<vector<int>>&matrix,int num){
        int cnt=0;
        int row=matrix.size();
        int col=matrix[0].size();
        int i=0,j=col-1;
        while(i<row&&j>=0){
            if(num>=matrix[i][j]){//如果大于该行的最大（最右）的元素，那么该元素之前的所有数都不大于num
                cnt+=j+1;
                i++;
            }
            else{//否则往左移一个元素
                j--;
            }
        }
        return cnt;
    }
};

668：

class Solution {
public:
    int findKthNumber(int m, int n, int k) {
        if(k==1) return 1;
        if(k==m*n) return m*n;
        int left=1,right=m*n,mid;
        while(left<right){
            mid=left+(right-left)/2;
            int temp=fun(m,n,mid);
            if(temp<k){
                left=mid+1;
            }
            else{
                right=mid;
            }
        }
        return left;
    }
    int fun(int m,int n,int num){
        int cnt=0;
        for(int i=1;i<=m;i++){
            cnt+=min(n,num/i);
        }
        return cnt;
    }
};