24.leetcode Merge k Sorted Lists(hard)[归并k个有序链表]

其实整体思想和归并k个有序数组是一样的,有三种方法,效率最慢的是两个有序链表和并之后再与其他链表合并,需要合并n-1次。第二种方法是两两一组分别归并,第三种方法是采用一个k大小的堆,刚开始的时候将每个链表的第一个元素都放入k的最小堆中,那么就可以直接得到k中最小的值,再将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* l1, ListNode* l2) {
        ListNode* p = l1;
        ListNode* q = l2;
        ListNode* newl = new ListNode(0);
        ListNode* temp = newl;
        while(p!=NULL && q != NULL)
        {
            if(p->val<q->val)
            {
                 newl->next = new ListNode(p->val);
                 newl = newl->next;
                 p = p->next;
            }           
            else
            {
                newl->next = new ListNode(q->val);
                newl = newl->next;
                q = q->next;
            }
        }
        if(p != NULL)
            newl->next = p;
        if(q != NULL)
            newl->next = q;
        return temp->next; 
    }
    /*ListNode* mergeKLists(vector<ListNode*>& lists) {
        int n = lists.size();
        if(n<=0) return NULL;
        if(n == 1) return lists[0];
        ListNode* first = lists[0];
        for(int i=1;i<n;i++)
        {
            first = mergeTwoLists(first,lists[i]);
        }
        return first;
    }*/
     ListNode* mergeKLists(vector<ListNode*>& lists) {
        int n = lists.size();
        if(n<=0) return NULL;
        if(n == 1) return lists[0];
        vector<ListNode*>::iterator mid = lists.begin()+n/2;
        vector<ListNode*> fhalf(lists.begin(),mid);
        cout<<"1:"<<fhalf.size()<<endl;
        vector<ListNode*> ahalf(mid,lists.end());
        cout<<"2:"<<ahalf.size()<<endl;
        ListNode* list1 = mergeKLists(fhalf);
        ListNode* list2 = mergeKLists(ahalf);
        //return list1;
        return mergeTwoLists(list1,list2);
        
    }
};


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值