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

本文介绍了一种在排序矩阵中寻找第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;
    }
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值