378. 有序矩阵中第K小的元素

378. 有序矩阵中第K小的元素

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

 

示例：

matrix = [
   [ 1,  5,  9],
   [10, 11, 13],
   [12, 13, 15]
],
k = 8,

返回 13。

 

提示：
你可以假设 k 的值永远是有效的，1 ≤ k ≤ n2 。

 

思路：这道题可以用二分法求解，效率会比较高。

我使用的是直接将二位数组转换成一维数组，然后排序，直接输入第k个数。

AC代码：

class Solution {

public:

    int kthSmallest(vector<vector<int>>& matrix, int k) {

        vector<int> v; 

        for (int i = 0; i < matrix.size(); i++) 

        {

            for (int j = 0; j < matrix[0].size(); j++) 

            {

                v.push_back(matrix[i][j]);

            }

        }

        sort(v.begin(), v.end());

        return v[k - 1];

    }

};