排序矩陣中的從小到大第k個數

題目:

在一個排序矩陣中找從小到大的第 k 個整數。
排序矩陣的定義爲:每一行遞增,每一列也遞增。code

樣例:

樣例
給出 k = 4 和一個排序矩陣:
[
[1 ,5 ,7],
[3 ,7 ,8],
[4 ,8 ,9],
]
返回 5。排序

思路:

將矩陣中每一個元素放入一個向量中,而後對向量元素進行排序,而後取出第k大的數。it

參考答案:

class Solution {
public:
    /*
     * @param matrix: a matrix of integers
     * @param k: An integer
     * @return: the kth smallest number in the matrix
     */
    int kthSmallest(vector<vector<int>> matrix, int k) {
        // write your code here
        if(matrix.empty())   return 0;
        int m = matrix.size();
        int n = matrix[0].size();
        vector<int> res;
        for(int i=0; i<m; i++){
            for(int j=0; j<n; j++){
                res.push_back(matrix[i][j]);
            }
        }
        sort(res.begin(),res.end());
        return res[k-1];
    }
};
相關文章
相關標籤/搜索