腾讯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位的,那么又该怎么判断呢?这个我真的没有考虑好。