Merge k Sorted Lists（C++合并K个升序链表）

最新推荐文章于 2022-08-15 17:09:21 发布

coolsunxu

最新推荐文章于 2022-08-15 17:09:21 发布

阅读量106

点赞数

分类专栏： C++ LeetCode 文章标签：链表 c++

本文链接：https://blog.csdn.net/coolsunxu/article/details/118187240

版权

C++ 同时被 2 个专栏收录

735 篇文章 17 订阅

订阅专栏

LeetCode

459 篇文章 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* helper(ListNode *head1,ListNode *head2) {
        ListNode *n_head=new ListNode(0);
        ListNode *h_head=n_head;
        while(head1!=nullptr && head2!=nullptr) {
            if(head1->val<head2->val) {
                ListNode *temp=head1->next;
                h_head->next=head1;
                head1=temp;
            } else if(head1->val>head2->val) {
                ListNode *temp=head2->next;
                h_head->next=head2;
                head2=temp;
            } else {
                ListNode *temp=head1->next;
                ListNode *temp1=head2->next;
                h_head->next=head1;
                h_head=h_head->next;
                h_head->next=head2;
                head1=temp;
                head2=temp1;
            }
            h_head=h_head->next;
        }

        if(head1!=nullptr) {
            h_head->next=head1;
        } else if(head2!=nullptr) {
            h_head->next=head2;
        } else h_head->next=nullptr;
        return n_head->next;
    }
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        ListNode *head=nullptr;
        for(int i=0;i<lists.size();i++) {
            head=helper(head,lists[i]);
        }

        return head;
    }
};