腾讯2020届实习笔试编程题3

腾讯2020届实习笔试编程题3

今晚刚刚完成了腾讯2020届的实习笔试(岗位:算法岗)，话不多说，现附上第三题编程题：

• 编程题3
  
• 我的解答

import numpy as np

def func(n, k, my_array):
    if 1<= n <= 105 and 1 <= k <= 105:
        flag = True
    else:
        return False

    output = []
    if flag:
        new_array = my_array
        for i in range(k):
            new_index = np.nonzero(new_array)
            new_array = new_array[new_index]
            if len(new_array) <= 1:
                output.append(0)
            else:
                min_value = new_array.min()
                new_array = new_array - min_value
                output.append(min_value)

    return output

if __name__ == '__main__':
    n, k = map(int, input().split())

    nlist = input().split()
    tmp = nlist[0]
    for i in range(1, len(nlist)):
        tmp = tmp + nlist[i]
    nlist = list(tmp)
    for i in range(len(nlist)):
        nlist[i] = int(nlist[i])
    my_array = np.array(nlist)
    result = func(n, k, my_array)

    for i in range(len(result)):
        print(result[i])

后来时间来不及了就没有在系统上测试，虽然基本思路有了，但是明显有一个问题在于：
第二行输入n个数字，每两个数字之间一个空格，但是我只考虑了每个数字为个位数的情况，如果输入的数字有超过2位的，那么又该怎么判断呢？这个我真的没有考虑好。