Kotlin Leetcode - 977. Squares of a Sorted Array

class Solution {
    fun sortedSquares(nums: IntArray): IntArray {
        
    }
}

解題思路

這一題考的是對陣列的處理

我們可以用 map() 的方式

來處理陣列內的每個元素

之後用 sorted() 進行排序

點擊展開解答

class Solution {
    fun sortedSquares(nums: IntArray): IntArray = nums
        .map { it * it }
        .sorted()
        .toIntArray()
}

如果你不想要做 toIntArray()

可以將回傳的型態轉換成 List<Int>

class Solution {
    fun sortedSquares(nums: IntArray): List<Int> = nums
        .map { it * it }
        .sorted()
}

或者直接省略型別，讓 Kotlin 的編譯器自行判斷

class Solution {
    fun sortedSquares(nums: IntArray) = nums
        .map { it * it }
        .sorted()
}