05 May 2023•Python

Sort Colors

Python solution for Sort Colors.

The Sort Colors solution is below.

Solution files

Python•sort-colors/solution.py

from typing import List


class Solution:
    def sortColors(self, nums: List[int]) -> None:
        left = 0
        current = 0
        right = len(nums) - 1

        while current <= right:
            color = nums[current]

            if color == 0:
                nums[left], nums[current] = nums[current], nums[left]
                left += 1
                current += 1
                continue

            if color == 2:
                nums[right], nums[current] = nums[current], nums[right]
                right -= 1
                continue

            current += 1