牛客网：NC51 合并k个已排序的链表

最新推荐文章于 2025-11-30 16:20:41 发布

原创最新推荐文章于 2025-11-30 16:20:41 发布 · 202 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#链表 #数据结构

算法同时被 2 个专栏收录

278 篇文章

订阅专栏

菜鸟刷题记录

204 篇文章

订阅专栏

这篇博客探讨了如何使用Java实现合并K个有序链表的问题。首先介绍了暴力方法，即逐个比较节点并合并，但这种方法时间复杂度过高。然后引入了分治策略，通过递归将问题拆分为更小的部分，有效降低了时间复杂度，提高了算法效率。

使用链表结合，暴力的方法

import java.util.*;
/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) {
 *         val = x;
 *         next = null;
 *     }
 * }
 */
public class Solution {
    public ListNode mergeKLists(ArrayList<ListNode> lists) {
        ListNode node = null;
        for (int i = 0; i < lists.size(); i++){
            node = merge(node, lists.get(i));
        }
        return node;
    }
    
    private ListNode merge(ListNode node1, ListNode node2){
        if (node1 == null) return node2;
        if (node2 == null) return node1;
        if (node1.val > node2.val) {
            node2.next = merge(node1, node2.next);
            return node2;
        }
        node1.next = merge(node1.next, node2);
        return node1;
    }
}

不行，时间复杂度过高。

在一的基础上添加分治，减少时间复杂度

import java.util.*;
/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) {
 *         val = x;
 *         next = null;
 *     }
 * }
 */
public class Solution {
    public ListNode mergeKLists(ArrayList<ListNode> lists) {
        return divideMerge(lists, 0 , lists.size() - 1);
    }
    
    private ListNode divideMerge(ArrayList<ListNode> lists, int left, int right){
        if (left > right){
            return null;
        }
        // 找到中间节点
        if (left == right){
            return lists.get(left);
        }
        int mid = (left + right) >> 1;
        return merge(divideMerge(lists, left, mid ), divideMerge(lists, mid + 1, right));
    }
    
    private ListNode merge(ListNode node, ListNode node2){
        if (node == null) return node2;
        if (node2 == null) return node;
        if (node.val > node2.val) {
            node2.next = merge(node, node2.next); 
            return node2;
        }
        node.next = merge(node.next, node2);
        return node;
    }
}