字节跳动算法-顺序打印m个长度为n的数组
思路
维护一个长度为n的数组,分别表示各个数组走到了哪里,每次找最小的一个数。每找出来一个数需要比较m次,需要找m*n次,时间复杂度是 m^2xN
代码
def printNSortArray(self,a):
if not a:
return
if not a[0]:
return
m = len(a)
n = len(a[0])
index = [0]*m
for i in range(m*n):
min_value = 2 ** 31
min_i = 0 # 最小数的在第几个数组中
for i,value in enumerate(index):
if value > n - 1: # 如果长度超过n-1了,说明这个数组走到头了
continue
if a[i][value] < min_value:
min_value = a[i][value]
min_i = i
print(min_value)
index[min_i] += 1# 向下走一位