在一個排序矩陣中找從小到大的第 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]; } };