-
Notifications
You must be signed in to change notification settings - Fork 614
/
75.py
30 lines (26 loc) · 989 Bytes
/
75.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
'''
Given an array with n objects colored red, white or blue, sort them in-place so that objects of the same color are adjacent, with the colors in the order red, white and blue.
Here, we will use the integers 0, 1, and 2 to represent the color red, white, and blue respectively.
Note: You are not suppose to use the library's sort function for this problem.
Example:
Input: [2,0,2,1,1,0]
Output: [0,0,1,1,2,2]
'''
class Solution(object):
def sortColors(self, nums):
"""
:type nums: List[int]
:rtype: void Do not return anything, modify nums in-place instead.
"""
zero, last = 0, len(nums)-1
index = 0
while index <= last:
if nums[index] == 1:
index += 1
elif nums[index] == 0:
nums[index], nums[zero] = nums[zero], nums[index]
index += 1
zero += 1
elif nums[index] == 2:
nums[last], nums[index] = nums[index], nums[last]
last -= 1