[Leetcode] Merge k Sorted Lists

最新推荐文章于 2024-04-03 21:24:25 发布

原创最新推荐文章于 2024-04-03 21:24:25 发布 · 353 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

36 篇文章

订阅专栏

本文详细介绍了如何将多个已排序的链表合并为一个有序的链表，通过递归调用合并两个链表的方法，实现整体链表的有序合并。代码示例清晰展示了算法实现过程。

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

感觉就是上一题的重复版，合并k次即可。

ListNode *mergeTwoLists(ListNode *l1, ListNode *l2) 
{
    if(!l1) return l2;
    if(!l2) return l1;
    ListNode * head=l1;					//把1一直当做新头结点
    ListNode * pre=NULL;
    while(l1 !=NULL && l2 !=NULL)
    {
    	if( l2->val < l1->val )				 //2比1小，就要插到前面
    	{
    		ListNode *tmp=l2->next;		
    		l2->next=l1;
    		if(pre ==NULL)  head=l2;		//如果1是头结点，更新头
    		else pre->next=l2;		        //加入2
    		pre=l2;				        //1后移一位
    		l2=tmp;					//2后移一位
    	}
    	else
    	{	
    		pre=l1;	
    		l1=l1->next;				//2的点比1的大，直接1右移
    	}
    }
    
    if(l1 ==NULL && l2 !=NULL)		//跳出循环说明2后面的都比1大，直接接在后面
    	pre->next=l2;
    
    return head;
}

ListNode *mergeKLists(vector<ListNode *> &lists) 
{  
    if(lists.size() == 0) return NULL;  
    ListNode *p = lists[0];  
    for(int i=1; i<lists.size(); i++){  
        p = mergeTwoLists(p, lists[i]);  
    }  
    return p;  
}