Solution: Dutch National Flag Problem

def dutch_national_flag(lst):
    """
    This function uses selection_sort approach to solve Dutch National Flag Problem
    :param lst: A list of integers
    :return: A list of solved Dutch National Flag Problem
    """
    size = len(lst)
    index = 0
    for i in range(size):
        index = find_min(lst, i, size)
        lst[i], lst[index] = lst[index], lst[i]
    return lst
def find_min(lst, start, end):
    """
    Finds the minimum value index in a list in specific range
    :param lst: A list of integers
    :param start: Starting index
    :param end: Ending index
    :return: Minimum value index
    """
    min = lst[start]
    index = start
    for x in range(start, end):
        if lst[x] < min:
            min = lst[x]
            index = x
    return index
# Driver to test above code
if __name__ == '__main__':
    lst = [2, 0, 0, 1, 2, 1]
    print(dutch_national_flag(lst))