Kotlin Leetcode - 832. Flipping an Image
題目連接:https://leetcode.com/problems/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 列表