Kotlin Leetcode - 832. Flipping an Image

題目連接

class Solution {
    fun flipAndInvertImage(image: Array<IntArray>): Array<IntArray> {

    }
}

解題思路

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

邏輯本身很簡單

我們可以用 Array 支援的 forEach 和 forEachIndexed

讓這段邏輯寫起來更簡單

Kotlin 參考解答

forEach 的寫法如下

class Solution {
    fun flipAndInvertImage(image: Array<IntArray>): Array<IntArray> {
        image.forEach {
            for (i in 0 until it.size / 2) {
                val tmp = it[i]
                it[i] = it[it.size - i - 1]
                it[it.size - i - 1] = tmp
            }
        }
        image.forEach {
            it.forEachIndexed { i, _ ->
                it[i] = 1 - it[i]
            }
        }
        return image
    }
}

---

回到 leetcode 列表