ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

LeetCode 23. 合并K个升序链表(分治 / 堆)

2021-01-18 17:31:45  阅读:167  来源: 互联网

标签:ListNode val 23 lists next 链表 l2 l1 升序


在这里插入图片描述
思路:
分治:分为两部分分别求解,再把求解结果按照两个链表合并的方式合并。
堆:将所有的节点放入堆中,每次取最小权值的节点。

分治:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode*merge(ListNode*l1,ListNode*l2) {
        if(l1 == NULL) return l2;
        if(l2 == NULL) return l1;
        ListNode* head = new ListNode();
        ListNode* h = head;
        while(l1 && l2) {
            if(l1 -> val <= l2 -> val) {
                h -> next = l1;
                l1 = l1 -> next;
            } else {
                h -> next = l2;
                l2 = l2 -> next;
            }
            h = h -> next;
        }
        if(l1 != NULL) {
            h -> next = l1;
        } else if(l2 != NULL) {
            h -> next = l2;
        }
        return head -> next;
    }
    ListNode*dfs(int l,int r,vector<ListNode*>&lists) {
        if(l > r) return NULL;
        if(l == r) return lists[l];
        int mid = (l + r) >> 1;
        return merge(dfs(l,mid,lists),dfs(mid + 1,r,lists));
    }
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        return dfs(0,lists.size() - 1,lists);
    }
};

堆:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        ListNode*head = new ListNode();
        ListNode*h = head;
        struct cmp {
            bool operator()(ListNode*l1,ListNode*l2) {
                return l1 -> val > l2 -> val;
            }
        };
        priority_queue<ListNode*,vector<ListNode*>,cmp>q;
        for(int i = 0;i < lists.size();i++) {
            if(lists[i] != NULL) q.push(lists[i]);
        }
        while(!q.empty()) {
            ListNode*now = q.top();q.pop();
            h -> next = now;
            h = h -> next;
            if(now -> next != NULL) 
                q.push(now -> next);
        }
        return head -> next;
    }
};

标签:ListNode,val,23,lists,next,链表,l2,l1,升序
来源: https://blog.csdn.net/tomjobs/article/details/112789213

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有