P69_2.kt

package ch17

class P69_2 {
    fun sortColors(nums: IntArray) {
        // red와 blue는 양쪽 끝 포인터이며, white는 중앙에서 움직이는 포인터다.
        var red = 0
        var white = 0
        var blue = nums.size

        // 두 값이 역전될 때까지 반복
        while (white < blue) {
            // white 포인터가 중앙값보다 작다면
            if (nums[white] < 1) {
                // red와 white 스왑
                nums[red] = nums[white].also { nums[white] = nums[red] }
                // red와 white 포인터는 한 칸씩 우측으로 이동
                red++
                white++
            } else if (nums[white] > 1) { // white 포인터가 중앙값보다 크다면
                // blue 포인터 좌측으로 이동
                blue--
                // white와 blue 스왑
                nums[white] = nums[blue].also { nums[blue] = nums[white] }
            } else {
                // 값이 같은 경우이므로, 스왑 없이 white 포인터만 우측으로 이동
                white++
            }
        }
    }
}