Advertisement
Guest User

Untitled

a guest
Oct 18th, 2019
127
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.85 KB | None | 0 0
  1. class Solution:
  2. def sortColors(self, nums: List[int]) -> None:
  3. """
  4. Do not return anything, modify nums in-place instead.
  5. """
  6. n = [0] * 3
  7. for num in nums:
  8. n[num] += 1
  9. for i in range(n[0]): nums[i] = 0
  10. for i in range(n[0], n[0] + n[1]): nums[i] = 1
  11. for i in range(n[0] + n[1], n[0] + n[1] + n[2]): nums[i] = 2
  12.  
  13. '''
  14. class Solution:
  15. def sortColors(self, nums: List[int]) -> None:
  16. """
  17. Do not return anything, modify nums in-place instead.
  18. """
  19. i, i0, i2 = 0, 0, len(nums) - 1
  20. while i <= i2:
  21. if nums[i] == 0:
  22. nums[i], nums[i0] = nums[i0], nums[i]
  23. i0 += 1
  24. elif nums[i] == 2:
  25. nums[i], nums[i2] = nums[i2], nums[i]
  26. i -= 1
  27. i2 -= 1
  28. i += 1
  29. '''
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement