链表合并问题

1 合并有序链表问题

1.1 合并两个有序链表

思路大概是创建一个新链表，谁大就把谁接入到新链表上面去。
思路很简单，实操的时候有许多细节和分类讨论，同时如何优化代码也是很重要的。
先上代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* mergeTwoLists(ListNode* list1, ListNode* list2) {
        ListNode* new_head = new ListNode();
        ListNode* res = new_head;

        while(list1!=NULL && list2!= NULL)
        {
            if(list1->val >= list2->val)
            {
                res->next = list2;
                list2 = list2->next;
            }
            else
            {
                res->next = list1;
                list1 = list1->next;
            }
            res = res->next;
        }
        
        res->next = list1 == NULL ? list2 : list1;
        return new_head->next;
    }
};

首先是遍历两个链表，逐个插入，相等的情况下用else可以在下一次循环中成功处理。而最后某个链表遍历到头之后，可以直接接上未遍历完的链表。

1.2 合并k个有序链表

承接上面的问题，简化为多次合并两个链表即可解决问题。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        ListNode* res = NULL;
        for(auto it = lists.begin(); it != lists.end(); it++)
        {
            res = mergeTwoLists(res, *it);
        }
        return res;

    }
    ListNode* mergeTwoLists(ListNode* list1, ListNode* list2) {
        ListNode* new_head = new ListNode();
        ListNode* res = new_head;

        while(list1!=NULL && list2!= NULL)
        {
            if(list1->val >= list2->val)
            {
                res->next = list2;
                list2 = list2->next;
            }
            else
            {
                res->next = list1;
                list1 = list1->next;
            }
            res = res->next;
        }
        
        res->next = list1 == NULL ? list2 : list1;
        return new_head->next;
    }
};