23. Merge k Sorted Lists (妙啊)

这道题是那道 Merge two sorted lists的升级版,这次要求对多个链表进行排序。如果用链表来实现的话,可能会比较繁琐。这里我用了比较投机的一种方法,先用一个vector把所有链表的val存下来,再排序,最后再生成一个链表。这样写既简单,速度又快,不过可能偏离了这道题的本意了吧~

class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
    	vector<int> values;
        for (int i = 0; i < lists.size(); ++i)
        {
        	ListNode * p = lists[i];
        	while(p != NULL){
        		values.push_back(p->val);
        		p = p->next;
        	}
        }
        sort(values.begin(), values.end());
        ListNode * head = new ListNode(0);
		ListNode * p = head;
		for (int i = 0; i < values.size(); ++i)
		{
			ListNode * temp = new ListNode(values[i]);
			p->next = temp;
			p = p->next;
		}
		if(head == NULL)cout << 1;
		head = head->next;

		return head;
    }
};


这里贴出答案里利用Merge two sorted lists来解决的版本,也很巧妙。

class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(lists.empty()){
            return nullptr;
        }
         while(lists.size() > 1){
            lists.push_back(mergeTwoLists(lists[0], lists[1]));
            lists.erase(lists.begin());
            lists.erase(lists.begin());
        }
        return lists.front();
    }
    ListNode *mergeTwoLists(ListNode *l1, ListNode *l2) {
    if(l1 == nullptr){
        return l2;
    }
    if(l2 == nullptr){
        return l1;
    }
    if(l1->val <= l2->val){
        l1->next = mergeTwoLists(l1->next, l2);
        return l1;
    }
    else{
        l2->next = mergeTwoLists(l1, l2->next);
        return l2;
    }
}
};


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值