合併 K 個排序序列 題目鏈接
合併 k 個排序鏈表,返回合併後的排序鏈表。請分析和描述算法的複雜度。
示例:
輸入:
[
1->4->5,
1->3->4,
2->6
]
輸出: 1->1->2->3->4->4->5->6
解題思路:採用分治的思想,將 K 個鏈表的合併問題轉換成,合併 2 個有序鏈表的問題
typedef struct ListNode List; // 定義一個 struct ListNode 類型 名爲 List
List *mergeTwoLists(List *list1, List *list2)
{
if (!list1&&!list2) return NULL;//當兩條鏈表都爲遍歷結束則返回 NULL
else if (!list1) return list2;//因爲都是有序則一條遍歷結束後只要將另一條鏈表接在後面即可
else if (!list2) return list1;
else if (list1->val<list2->val)
{
list1->next = mergeTwoLists(list1->next, list2);
return list1;
}
else
{
list2->next = mergeTwoLists(list1, list2->next);
return list2;
}
}
struct ListNode* mergeKLists(struct ListNode** lists, int listsSize)
{
if (listsSize == 0) return NULL;
else if (listsSize == 1) return lists[0];
else if (listsSize == 2) return mergeTwoLists(lists[0], lists[1]);
return mergeTwoLists(mergeKLists(lists, listsSize / 2),
mergeKLists(lists + listsSize / 2, listsSize - listsSize / 2));//將 K 個鏈表分解成 2 個鏈表的組合
}