23. Merge k Sorted Lists

题目描述:

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

Example:

Input:
[
  1->4->5,
  1->3->4,
  2->6
]
Output: 1->1->2->3->4->4->5->6

代码:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    int findMini(vector<ListNode*>& lists){
        int mini = lists[0] -> val;
        for(int i = 1; i < lists.size(); i++){
            ListNode* tmp = lists[i];
            int num = tmp -> val;
            if(num < mini) mini = num;
        }
        return mini;
    }
    void initial(vector<ListNode*>& lists){
        for(int i = 0; i < lists.size(); i++){
            if(lists[i] == NULL){
                lists.erase(lists.begin() + i);
                i--;
            }    
        }
    }
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        initial(lists);
        ListNode* result = new ListNode(0);
        ListNode* current = result;
        while(lists.size() != 0){
            int mini = findMini(lists);
            for(int i = 0; i < lists.size(); i++){
                while(lists[i] -> val == mini){
                    ListNode* node = new ListNode(mini);
                    current -> next = node;
                    current = current -> next;
                    lists[i] = lists[i] -> next;
                    if(lists[i] == NULL){
                        lists.erase(lists.begin() + i);
                        i--;
                        break;
                    }
                }
            }    
        }
        return result -> next;
    }
};

昨天七夕,浪着浪着就忘记做了(毕竟刚开始0.0),这个算补昨天的吧。本题是一个hard题,AC率有点吓人,才百分之29点多,但是我感觉题目稍微有点问题,在题目开始的结构体定义中,构造函数是必须有一个初始参数int x的,也就是不存在空链表这么一说,但是在我提交了第一次的时候,收到一个WA,原因是当input为[[]]时候,output为[],也就是说本题的输入可以为一个size不为0的,但是链表可以为NULL的vector。要不是这个小bug,博主一次AC了有木有!!!!*★,°*:.☆( ̄▽ ̄)/$:*.°★* 。

好了,废话不多说了,来看解题思路,题目要求合并k个有序链表,并且返回一个有序链表。我的思路,每次找所有链表头结点val的最小值。然后遍历vector,对于vector中的每个链表,进行遍历,如果结点值等于mini,则result链表中添加val为mini的结点,并且将该链表的头结点删除,遍历直到头结点的值不等于mini。或者当该链表为NULL时,将其从vector中erase掉。

findMini函数即每次从所有链表的头结点中找最小值。initial函数就是我开始说的,此题存在空链表,所以通过initial函数将这些空链表直接删除。

静心尽力!

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值