Question

How to sort a list of 0s, 1s and 2s in Python

Answer 1

def sort012List(lst) :
    lo = 0
    curr = 0
    hi = len(lst) - 1
    while (curr <= hi) :
        if lst[curr] == 0 :
            lst[lo], lst[curr] = lst[curr], lst[lo]
            lo += 1
            curr += 1
        elif lst[curr] == 1 :
            curr += 1
        elif lst[curr] == 2 :
            lst[curr], lst[hi] = lst[hi], lst[curr]
            hi -= 1
    return lst
        
        

lst = [1, 2, 2, 0, 1, 1, 0, 2, 0, 1, 0, 0, 1]

lst = sort012List(lst)

print(lst)



'''
run:

[0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 2, 2, 2]

'''

 

70+ SQL courses for beginners and professionals