leetcode - kth smallest element in sorted matrix(kotlin)

silver·2021년 7월 8일
0

level - medium

[문제 내용]
주어진 배열을 오름차순으로 정렬하여 k번째에 있는 값을 반환

[example 1]]

Input: matrix = [[1,5,9],[10,11,13],[12,13,15]], k = 8
Output: 13
Explanation: The elements in the matrix are [1,5,9,10,11,12,13,13,15], and the 8th smallest number is 13

[example 2]

Input: matrix = [[-5]], k = 1
Output: -5

[해결 방법]
문제가 너무 쉬운관계로 설명은 생략한다.

class Solution {
    fun kthSmallest(matrix: Array<IntArray>, k: Int): Int {
        val mat = IntArray(matrix.size * matrix[0].size)
        var index = 0
        for(m in matrix) {
            for(_m in m) {
                mat[index++] = _m
            }
        }

        Arrays.sort(mat)
        return mat[k-1]
    }
}

0개의 댓글