字节跳动算法-顺序打印m个长度为n的数组

字节跳动算法-顺序打印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# 向下走一位