LeetCode. 23 - 合併 K個排序序列

合併 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 個鏈表的組合
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章