合并K个已排序的链表

最新推荐文章于 2013-05-26 14:56:31 发布

chunxia75qin

最新推荐文章于 2013-05-26 14:56:31 发布

阅读量561

点赞数

分类专栏： c++ code 文章标签： LeetCode 算法链表

本文链接：https://blog.csdn.net/chunxia75qin/article/details/8843452

版权

c++ code 专栏收录该内容

60 篇文章 0 订阅

订阅专栏

LeetCode上的题：

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

分析：合并K个已排序的链表，实际上就是重复合并两个已排序的链表。

代码如下：

//Definition for singly-linked list.
struct ListNode {
     int val;
     ListNode *next;
     ListNode(int x) : val(x), next(NULL) {}
};

class Solution {
public:
    ListNode *mergetwoLists(ListNode * list1,ListNode * list2){
        ListNode * head1=NULL,*head2=NULL,*head=NULL;
        if(list1==NULL)return list2;
        if(list2==NULL)return list1;
        if(list1->val<=list2->val)
        {
            head=head1=list1;
            head2=list2;
        }
        else
        {
            head=head1=list2;
            head2=list1;
        }
        while(head2!=NULL)
        {
            while(head1->next!=NULL&&head1->next->val<=head2->val)
            {
                head1=head1->next;
            }
            if(head1->next==NULL)
            {
                head1->next=head2;
                break;
            }
            else
            {
                ListNode * tmp1=head1->next;
                ListNode * tmp2=head2->next;
                head1->next=head2;
                head2->next=tmp1;
                head1=head2;
                head2=tmp2;
            }
        }
        return head;
    }
    ListNode *mergeKLists(vector<ListNode *> &lists) {
       // if(lists.size()<=0)return NULL;
       // if(lists.size()==1)return lists[0];
        ListNode *head=NULL;
        for(int i=0;i<lists.size();i++)
        {
            head=mergetwoLists(lists[i],head);
        }
        return head;
    }
};