Sort Colors

Sizga n ta elementdan iborat nums massivi berilgan bo'lib, ular qizil, oq yoki ko'k ranglarga bo'yalgan. Ular bir xil rangdagi elementlar yonma-yon kelishi uchun (qizil, oq va ko'k ketma-ketligida) ularni in-place (joyida) tartiblang.

Biz ranglarni ifodalash uchun mos ravishda 0, 1 va 2 butun sonlaridan foydalanamiz.

Siz standart kutubxonadagi tayyor sort funksiyasidan foydalanmasligingiz kerak. Eslatma: Bu yerda bizning test tizimimiz talabi bilan funksiya massivni o'zgartiribgina qolmay, o'sha o'zgartirilgan massivni qaytarishi (return) kerak.

func solve(nums []int) []int {
    // ...
}

Misollar

Input: nums = [2,0,2,1,1,0]
Output: [0,0,1,1,2,2]

Input: nums = [2,0,1]
Output: [0, 1, 2]

Input: nums = [0]
Output: [0]

Maslahatlar

1. Uchta qiymat (0,1,2) borligi sababli uchta ko'rsatkich bilan massivni bir marta aylanib chiqadigan usulni (Dutch National Flag algoritmi) o'ylab ko'ring.
2. low, mid, high ko'rsatkichlarini yuriting: nums[mid]==0 bo'lsa uni low bilan almashtirib low va mid ni oshiring, nums[mid]==1 bo'lsa faqat mid ni oshiring, nums[mid]==2 bo'lsa uni high bilan almashtirib high ni kamaytiring (bu holda mid ni oshirmang, chunki yangi kelgan qiymat hali tekshirilmagan).