程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> 關於C++ >> Merge k Sorted Lists

Merge k Sorted Lists

編輯:關於C++

Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.


算法一,自底至頂遞歸。

時間復雜度為O(nlogk)。

空間復雜度為O(k)。

此算法在leetcode上實際執行時間為 144ms (130 test cases)。


/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode *mergeKLists(vector &lists) {
        if (lists.empty()) return 0;
        if (lists.size() == 1) return lists.back();

        vector l;
        while (!lists.empty()) {
                if (lists.size() == 1) {
                        l.push_back(lists.back());
                        lists.pop_back();
                }
                else {
                        ListNode *p = lists.back();
                        lists.pop_back();
                        ListNode *q = lists.back();
                        lists.pop_back();

                        l.push_back(merge(p, q));
                }
        }

        return mergeKLists(l);
    }

    ListNode *merge(ListNode *p,ListNode *q) {
        ListNode fake(0);
        ListNode *m = &fake;

        while (p && q) {
                if (p->val < q->val) {
                        m->next = p;
                        p = p->next;
                }
                else {
                        m->next = q;
                        q = q->next;
                }
                m = m->next;
        }

        m->next  = p ? p : q;
        return fake.next;
    }
};

算法二,待續

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved