题:
请给出一个时间为O(nlgk)、用来将k个已排序链表合并为一个排序链表的算法。此处n为所有输入链表中元素的总数。
(提示:用一个最小堆来做k路合并)
k个已排序链表,假设每一个链表的第一个元素为root(i) (i = 1,2,3...k)。
那么可以为这k个元素root(i)构造一个最小堆Heap_Min,其时间复杂度为O(lgk)。
此时堆顶元素假设为root(x),根据最小堆性质,root(x)是所有root(i)中的最小值。
而每个链表已经排序,所以其也是整个n个元素中的最小值,将其取出,放入另一个链表res中。
对于root(x)所在的链表,由于第一个元素已经被取出,所以用此链表中的第二个元素代替root(x),由于root(x)的值已经改变,整个最小堆Heap_Min需要调整,时间复杂度为O(lgk)。假如某个链表的值都被取空,那么就用Heap_Min的最后一个元素root(y)作为整个堆的新树根。依次,知道取出所以元素。
根据以上分析可知:每取出一个元素都要进行一次Heap_Min的堆调整,而调整的时间复杂度为O(lgk),一共要进行n次,总时间复杂度为O(nlgk)。