合并K个链表

思路:借鉴《算法4》上的自底向上的归并排序
1.从1开始,两两合并,然后四四合并,八八合并,直到链表数量为一

时间复杂度:O(Nlogk)
空间复杂度:O(1)

#define less(X,Y) ((X)<(Y)?(X):(Y))

void merge(struct ListNode** lists,int start,int mid,int end){
	//判断最后是否只剩一个链表,一个链表时不需要合并
	if(end<=mid)
		return ;
	
	//合并两个有序链表的操作	
	struct ListNode* head=(struct ListNode*)mallco(sizeof(struct ListNode));
	head->val=0;
	head->next=NULL;
	struct ListNode *tail=head;
	
	int fir=start,sec=mid+1;
	while(lists[fir]&&lists[sec]){
		if(lists[fir]->val<=lists[sec]->val){
			tail->next=lists[fir];
			lists[fir]=lists[fir]->next;
		}
		else{
			tail->next=lists[sec];
			lists[sec]=lists[sec]->next;
		}
		tail=tail->next;
	}

	if(lists[fir])
		tail->next=lists[fir];
	if(lists[sec])
		tail->next=lists[sec];	
	
	lists[start]=head->next;
	free(head);
}

struct ListNode* mergeKList(struct ListNode** lists,int listsSize){
	if(NULL==lists||listsSize<1)
		return NULL;
	for(int i=1;i<listsSize;i*=2){			//从一开始,逐渐加倍
		for(int j=0;j<listsSize;j+=2*i)		//j为第一个链表的索引(相对位置)
			merge(lists,j,j+i-1,less(j+2*i-1,listsSize-1));
	}
	return lists[0];
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值