leetcode 75

sort_colors.kt

class Solution {
    fun sortColors(nums: IntArray): Unit {
        var index = 0
        var redIdx = 0
        var blueIdx = nums.lastIndex
        
        while(index <= blueIdx) {
            when {
                nums[index] == 0 -> {
                    nums[redIdx] = nums[index].also {nums[index] = nums[redIdx]}
                    ++redIdx
                    ++index
                }
                nums[index] == 2 -> {
                    nums[blueIdx] = nums[index].also {nums[index] = nums[blueIdx]}
                    --blueIdx
                }
                else -> ++index
            }
        }
    }
}