Leetcode 23. Merge k Sorted Lists

版权声明:博客文章都是作者辛苦整理的,转载请注明出处,谢谢! https://blog.csdn.net/Quincuntial/article/details/81701799

文章作者:Tyan 博客:noahsnail.com | CSDN | 简书

1. Description

2. Solution

  • Version 1
/**
 * 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 size = lists.size();
        if(size == 0) {
            return NULL;
        }
        ListNode* result = lists[0];
        for(int i = 1; i < size; i++) {
            result = mergeTwoLists(result, lists[i]);
        }
        return result;
    }

private:
    ListNode* mergeTwoLists(ListNode* first, ListNode* second) {
        ListNode* ptrFirst = first;
        ListNode* ptrSecond = second;
        ListNode* head = new ListNode(0);
        ListNode* current = head;
        while(ptrFirst && ptrSecond) {
            if(ptrFirst->val > ptrSecond->val) {
                current->next = ptrSecond;
                ptrSecond = ptrSecond->next;
            }
            else {
                current->next = ptrFirst;
                ptrFirst = ptrFirst->next;
            }
            current = current->next;
        }
        if(ptrFirst) {
            current->next = ptrFirst;
        }
        if(ptrSecond) {
            current->next = ptrSecond;
        }
        return head->next;
    }

};
  • Version 2
/**
 * 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 size = lists.size();
        if(size == 0) {
            return NULL;
        }
        ListNode* result = NULL;
        while(lists.size() != 1) {
            ListNode * l1 = lists.back();
            lists.pop_back();
            ListNode * l2 = lists.back();
            lists.pop_back();
            result = mergeTwoLists(l1, l2);
            lists.insert(lists.begin(), result);
        }
        return lists[0];
    }

private:
    ListNode* mergeTwoLists(ListNode* first, ListNode* second) {
        ListNode* ptrFirst = first;
        ListNode* ptrSecond = second;
        ListNode* head = new ListNode(0);
        ListNode* current = head;
        while(ptrFirst && ptrSecond) {
            if(ptrFirst->val > ptrSecond->val) {
                current->next = ptrSecond;
                ptrSecond = ptrSecond->next;
            }
            else {
                current->next = ptrFirst;
                ptrFirst = ptrFirst->next;
            }
            current = current->next;
        }
        if(ptrFirst) {
            current->next = ptrFirst;
        }
        if(ptrSecond) {
            current->next = ptrSecond;
        }
        return head->next;
    }

};

Reference

  1. https://leetcode.com/problems/merge-k-sorted-lists/description/

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券