leetcode -- （1）按照指定長度保留數組數據

 在原題基礎上實現按照參數保留重複的數據

def remove_duplicates(array):
    if len(array) == 0:
        return 0
    last_index = 0
    for value in array:
        if value != array[last_index]:
            last_index += 1
            array[last_index] = value
    print(array[0:last_index+1])


def retain_duplicates(array, duplicates_count):
    if len(array) == 0:
        return 0
    last_index = 0
    current_index = 0
    for value in array:
        if current_index >= duplicates_count and value != array[last_index]:
            index = last_index + duplicates_count
            last_index += 1
            array[index] = value
        current_index += 1
    print(array[0:last_index + duplicates_count])


if __name__ == "__main__":
    remove_duplicates([3, 5, 5, 5, 6, 6, 77, 77, 77, 98, 98, 100, 100])
    retain_duplicates([3, 5, 5, 5, 6, 6, 77, 77, 77, 98, 100, 100, 105, 105, 105, 106], 2)