LeetCode题解之Merge k Sorted Lists 解法二

1、题目描述

2、分析

利用 vector 存储指针，同时合并k个链表。

 

3、代码

 ListNode* mergeKLists(vector<ListNode*>& lists) {
        
        vector<ListNode*> p;
        for (vector<ListNode*>::iterator it = lists.begin(); it != lists.end(); it++) {
            if (*it != NULL){
                ListNode *pn = *it;
                p.push_back(pn);
            }
        }
        
        if (p.size() == 0)
            return NULL;
        
        ListNode *dummy = new ListNode(0);
        ListNode *pn = dummy;
        
        while (p.size() > 0) {
            int n = minval(p);
            pn->next = p[n];
            pn = pn->next;
            p[n] = p[n]->next;
            if (p[n] == NULL)
                p.erase(p.begin() + n);
        }
        return dummy->next;
    }
    
    
    int minval(vector<ListNode*> p)
    {
        auto res = p.begin();
        for (auto it = p.begin(); it != p.end(); it++) {
            if ((*it)->val <= (*res)->val)
                res = it;
        }
        return res - p.begin();
    }