按照最朴素的方法,每轮都对所给列表进行一次遍历,O(n)的复杂度获得值最小的节点,并将其上的链表指针后移一位,一旦为空则剔除数组。数组为空时结束循环。
这样写时间复杂度较高,因为涉及到枚举最小值节点,数组的删除操作,但空间复杂度可以降到O(1)
class Solution {
public:
ListNode* mergeKLists(vector<ListNode*>& lists) {
int n = lists.size();
for(int i = n - 1; i >= 0; i--)
{
if(lists[i] == nullptr)
lists.erase(lists.begin() + i);
}
n = lists.size();
if(n == 0)
return nullptr;
ListNode* head = nullptr, *p = nullptr;
while(lists.size() > 0)
{
n = lists.size();
int minId = 0;
for(int i = 1; i < n; i++)
{
if(lists[i]->val < lists[minId]->val)
{
minId = i;
}
}
if(!head)
{
head = p = lists[minId];
}
else
{
p->next = lists[minId];
p = p->next;
}
lists[minId] = lists[minId]->next;
if(!lists[minId])
lists.erase(lists.begin() + minId);
}
return head;
}
};