def sortColors(nums):

    counts = [0, 0, 0]

    # Count the occurrences of 0s, 1s, and 2s

    for num in nums:

        counts[num] += 1

    # Update the original array with the sorted order

    i = 0

    for num in range(3):

        for _ in range(counts[num]):

            nums[i] = num

            i += 1

    return nums

​

# Example usage

arr = [2, 0, 2, 1, 1, 0]

print(sortColors(arr))

/* Dutch National flag algorithm  --  IN-PLACE */

        int low = 0 ,  mid = 0  , high = nums.size()-1 ; 

        while (mid <= high) {

            if (nums[mid] == 0) {

                swap(nums[low] , nums[mid]) ; 

                mid++ ; low++ ; 

            }

            else if (nums[mid] == 1) {

                mid++ ;

            }

            else {

                swap(nums[mid] , nums[high]) ; 

                high-- ; 

            }

        }