ひよっこゲームブログ

なにもかも初心者のひよっこがゆったりと何かする

Kotlinのソート

先週にLeetCodeのデイリーをやってて知らなかったことをメモ

Kotlinにソートがあったことを知らんかった
今まで使う機会なかったしなぁ

公式ドキュメントを見たら結構色々できることを知った

kotlinlang.org



Merge Sorted Array

Solution
Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.

The number of elements initialized in nums1 and nums2 are m and n respectively. You may assume that nums1 has a size equal to m + n such that it has enough space to hold additional elements from nums2.

Example 1:

Input: nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
Output: [1,2,2,3,5,6]

Example 2:

Input: nums1 = [1], m = 1, nums2 = [], n = 0
Output: [1]

検証時に書いたコード

fun merge(nums1: IntArray, m: Int, nums2: IntArray, n: Int): Unit {
        
        var numsArray: IntArray = intArrayOf()
        
        for(i in 0..m-1) {
            numsArray += nums1[i]
        }
        for(i in 0..n-1) {
            numsArray += nums2[i]
        }
               
        val result: Unit = numsArray.sort()
        println(numsArray.contentToString())
        
        return result
    }

fun main() {
    merge(intArrayOf(1,2,3,0,0,0), 3, intArrayOf(2,5,6), 3)
}

Result:

[1, 2, 2, 3, 5, 6]

本来はソートのアルゴリズムを書くべきなんだけどな・・・