leetcode -- （1）按照指定长度保留数组数据

2019独角兽企业重金招聘Python工程师标准>>>

 在原题基础上实现按照参数保留重复的数据

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)

 

转载于:https://my.oschina.net/u/3001485/blog/784837