/** * Definition for singly-linked list. * struct ListNode { * int val; * ListNode *next; * ListNode(int x) : val(x), next(NULL) {} * }; */ class Solution { public: ListNode* mergeKLists(vector<ListNode*>& lists) { int k = lists.size(); // 鏈表個數 if (k >= 2) { // 先將前兩個鏈表合併爲一個新鏈表,再把新鏈表依次和後面的鏈表合併 ListNode *head = mergeTwoLists(lists[0], lists[1]); for (int i = 2; i < k; i++) { head = mergeTwoLists(head, lists[i]); } return head; } else if (k == 1) { return lists[0]; } else { return NULL; } } ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) { ListNode *head = new ListNode(0); // 新建哨兵結點,方便操做 ListNode *temp = head; // 依次比較兩個鏈表的結點值,將值較小的結點插入到新建的鏈表後面 while(l1 && l2) { if (l2->val <= l1->val) { temp->next = l2; temp = temp->next; l2 = l2->next; } else { temp->next = l1; temp = temp->next; l1 = l1->next; } } // 其中一個鏈表比較完畢,將另一個鏈表剩餘結點直接插入到新建的鏈表後面 if (l1) { temp->next = l1; } else { temp->next = l2; } temp = head; head = head->next;// 刪除哨兵結點 delete(temp); return head; } };
/** * Definition for singly-linked list. * struct ListNode { * int val; * ListNode *next; * ListNode(int x) : val(x), next(NULL) {} * }; */ class Solution { public: ListNode* mergeKLists(vector<ListNode*>& lists) { int k = lists.size(); // 鏈表個數 if (k >= 2) { while(k != 1) // 循環合併過程直到剩餘一個鏈表 { if (k % 2 == 0) // 總共偶數個鏈表,依次合併相鄰兩個鏈表而後從前日後放到 vector 中去 { for (int i = 0; i < k; i=i+2) { lists[i / 2] = mergeTwoLists(lists[i], lists[i+1]); } k = k / 2; // 每次合併後的新鏈表個數 } else // 總共奇數個鏈表,依次合併相鄰兩個鏈表而後從前日後放到 vector 中去,最後餘一個鏈表直接放入 vector 最後面 { for (int i = 0; i < k - 1; i=i+2) { lists[i / 2] = mergeTwoLists(lists[i], lists[i+1]); } lists[k / 2] = lists[k-1]; k = k / 2 + 1; // 每次合併後的新鏈表個數 } } return lists[0]; } else if (k == 1) { return lists[0]; } else { return NULL; } } ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) { ListNode *head = new ListNode(0); // 新建哨兵結點,方便操做 ListNode *temp = head; // 依次比較兩個鏈表的結點值,將值較小的結點插入到新建的鏈表後面 while(l1 && l2) { if (l2->val <= l1->val) { temp->next = l2; temp = temp->next; l2 = l2->next; } else { temp->next = l1; temp = temp->next; l1 = l1->next; } } // 其中一個鏈表比較完畢,將另一個鏈表剩餘結點直接插入到新建的鏈表後面 if (l1) { temp->next = l1; } else { temp->next = l2; } temp = head; head = head->next;// 刪除哨兵結點 delete(temp); return head; } };
/** * Definition for singly-linked list. * struct ListNode { * int val; * ListNode *next; * ListNode(int x) : val(x), next(NULL) {} * }; */ class Solution { public: ListNode* mergeKLists(vector<ListNode*>& lists) { int k = lists.size(); struct cmp { bool operator()(ListNode* a, ListNode* b) { return a->val > b->val; } }; priority_queue<ListNode *, vector<ListNode*>, cmp> q; // 構建小頂堆 for (int i = 0; i < k; i++) { if (lists[i]) q.push(lists[i]); // 鏈表非空,加入隊列 } ListNode *head = new ListNode(0); // 新建哨兵結點,方便操做 ListNode *temp = head; while(!q.empty()) { temp->next = q.top(); // 將元素值最小的首結點插入到新鏈表後面 temp = temp->next; // 指針後移 q.pop(); // 首結點元素值最小的鏈表出隊 if (temp->next) // 把刪除首結點後的鏈表放入隊列中 q.push(temp->next); } temp = head; head = head->next; delete(temp); // 刪除哨兵結點 return head; } };
獲取更多精彩,請關注「seniusen」! spa