题解 | #合并k个已排序的链表#

合并k个已排序的链表

https://www.nowcoder.com/practice/65cfde9e5b9b4cf2b6bafa5f3ef33fa6

//分治思想运用在K个链表中方法  两个链表合并+分治
/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
  public:

    ListNode* Merge2(ListNode* pHead1, ListNode* pHead2) {
        if (pHead1 == NULL)
            return pHead2;
        if (pHead2 == NULL)
            return pHead1;
        ListNode* newList = new ListNode(0);
        ListNode* nHead;
        nHead = newList;
        while (pHead1 != NULL && pHead2 != NULL) {
            if (pHead1->val < pHead2->val) {
                newList->next = pHead1;
                pHead1 = pHead1->next;
                cout << "路过1";
            } else {
                newList->next = pHead2;
                pHead2 = pHead2->next;
                cout << "路过2";
            }
            newList = newList->next;
        }
        while (pHead1) {
            newList->next = pHead1;
            newList = newList->next;
            pHead1 = pHead1->next;
            cout << "路过3";
        }
        while (pHead2) {
            newList->next = pHead2;
            newList = newList->next;
            pHead2 = pHead2->next;
            cout << "路过4";
        }
        return nHead->next;
    }

    ListNode* merge1(vector<ListNode*>& lists, int left, int right) {
        if (left > right)
            return NULL;
        if (left == right)
            return lists[left];
        int mid = (left + right) / 2;
        return Merge2(merge1(lists, left, mid), merge1(lists, mid+1, right));

    }
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        return merge1(lists, 0, lists.size() - 1);
    }
};

全部评论

相关推荐

11-04 14:10
东南大学 Java
_可乐多加冰_:去市公司包卖卡的
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务