leetcode23. 合併K個排序鏈表_C++_hard

合併 k 個排序鏈表,返回合併後的排序鏈表。請分析和描述算法的複雜度。

示例:

輸入:
[
  1->4->5,
  1->3->4,
  2->6
]
輸出: 1->1->2->3->4->4->5->6

思路:是給兩個有序鏈表排序的變形,主要在於爲了提高速度,不要一個一個加上去,而是兩個變成一個長的,再兩個變成一個長的,知道最後成爲一個長子串。

 

C++代碼如下:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
    ListNode* Merge(ListNode* l1,ListNode* l2)
    {
        ListNode* start = new ListNode(-1);
        ListNode* end=start;
        while(l1 && l2)
        {
            if(l1->val < l2->val)
            {
                end->next=l1;
                l1=l1->next;
            }
            else
            {
                end->next = l2;
                l2=l2->next;
            }
            end = end->next;
        }
        if(l1)
            end->next = l1;
        else if(l2)
            end->next = l2;
        
        return start->next;
    }
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        int n=lists.size();
        if(n == 0) return {};
        while(n >1) {
            int k=(n+1)/2;
            for(int i=0;i<n/2;i++)
            {
                lists[i]=Merge(lists[i],lists[i+k]);
            }
            n=k;
        }
        return lists[0];
    }
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章