Dutch National Flag Problem

Solution.kt

class Solution {
    fun swapColor(nums: IntArray, k: Int, l: Int) {
        val tmp = nums[k]
        nums[k] = nums[l]
        nums[l] = tmp
    }

    fun sortColors(nums: IntArray) {
        var l = 0
        var m = 0
        var h = nums.size - 1
        while (m <= h) {
            when (nums[m]) {
                0 -> swapColor(nums, m++, l++)
                1 -> m++
                2 -> swapColor(nums, m, h--)
            }
        }
    }
}