[Algorithmus-Interviewfragen Python] K geordnete Listen zusammenführen

Titel: k geordnete Listen
zusammenführen Eingabe: [[3,4,5], [1,6], [2,6,7,8], [5,6,9]]
Ausgabe: [1, 2, 3, 4, 5, 5, 6, 6, 6, 7, 8, 9]
Idee: Weisen Sie jeder Liste einen Zeiger zu, vergleichen Sie die Elemente, auf die k Zeiger in jeder Schleife zeigen, nehmen Sie das kleinste und fügen Sie es in eine neue Liste ein , und Addiere 1 zum entsprechenden Zeiger, und die Bedingung für das Verlassen der Schleife ist, dass alle Elemente der Liste abgerufen wurden.
Code:

def merge(a):
    new = []
    idx = [0]*len(a) # k个指针
    while True:
        curmin = 1e6  
        curidx = 0
        flag = 0
        for i in range(len(idx)):
            if idx[i]<len(a[i]):
                if a[i][idx[i]]< curmin:
                    curmin = a[i][idx[i]]
                    curidx = i # 最小值在第curidx个列表
            else:
                flag += 1
        if flag == len(idx):
            break
        new.append(curmin)
        idx[curidx] += 1 # 最小值所在的列表的指针要+1
    return new