矩阵第K小

[编程题]矩阵第K小

热度指数：683 时间限制：C/C++ 1秒，其他语言2秒空间限制：C/C++ 256M，其他语言512M
算法知识视频讲解

给定一个 $n\times n\$ 的矩阵，每行每列都是升序排列的，请你找到矩阵中第 K 小的元素

数据范围： $1 \le n \le 500 \$

示例1

输入

[[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15],[16,17,18,19,20],[21,22,23,24,25]],5

输出

示例2

输入

[[1,1,1],[1,1,1],[1,1,1]],5

输出

算法知识视频讲解

小谷围做题家

package main
import "sort"

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param matrix int整型二维数组 
 * @param k int整型 
 * @return int整型
*/
func KthinMatrix( matrix [][]int ,  k int ) int {
    arr:=[]int{}
    for _,child:=range matrix{
        arr=append(arr,child...)
    }
    sort.Ints(arr)
    return arr[k-1]
}

发表于 2023-03-28 13:17:21 回复(0)