lintcode(401)排序矩阵中的从小到大第k个数

最新推荐文章于 2024-08-29 12:01:12 发布

原创最新推荐文章于 2024-08-29 12:01:12 发布 · 置顶 · 1.4k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #lintcode #简单算法

本文介绍了一种在排序矩阵中寻找第K小元素的方法。该矩阵的特点是每行和每列都按递增顺序排列。通过遍历并比较各行元素，最终找到目标值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

描述：

在一个排序矩阵中找从小到大的第 k 个整数。

排序矩阵的定义为：每一行递增，每一列也递增。

样例：

给出 k = 4 和一个排序矩阵：

[
  [1 ,5 ,7],
  [3 ,7 ,8],
  [4 ,8 ,9],
]

返回 5。

思路：

把原题分解成将若个有序数组的合并，建立数组记录各行已经的未合并元素的位置，直到循环执行k次为止

public class Solution {
    /**
     * @param matrix: a matrix of integers
     * @param k: an integer
     * @return: the kth smallest number in the matrix
     */
    public int kthSmallest(int[][] matrix, int k) {
        // write your code here
        if(matrix == null || matrix.length == 0){
            return 0;
        }
        int result = 0;
        int[] row = new int[matrix.length];
        for(int i = 0;i<row.length;i++){
            row[i] = 0;
        }
        for(int j = 0;j<k;j++){
            int flag = 0;
            int temp = Integer.MAX_VALUE;
            for(int i = 0;i<row.length;i++){
                if(row[i] < matrix[0].length){
                    if(matrix[i][row[i]] < temp){
                        temp = matrix[i][row[i]];
                        flag = i;
                    }    
                }
            }
            row[flag]++;
            result = temp;
        }
        return result;
    }
}