合并多个有序链表

1、维持一个最小堆,利用堆的性质每次从堆中取出一个最小值加入node中。

import java.util.ArrayList;
import java.util.List;

public class MergeKList {

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        ListNode node11 = new ListNode(1);
        ListNode node12 = new ListNode(3);
        ListNode node13 = new ListNode(5);
        ListNode node14 = new ListNode(7);
        node11.next = node12;
        node12.next = node13;
        node13.next = node14;

        ListNode node21 = new ListNode(1);
        ListNode node22 = new ListNode(2);
        ListNode node23 = new ListNode(3);
        node21.next = node22;
        node22.next = node23;

        ListNode node33 = new ListNode(3);
        ListNode node34 = new ListNode(4);
        node33.next = node34;

        ListNode list[] = { node11, node21, node33,null};
        ListNode head = mergeK(list);
        while (head != null) {
            System.out.println(head.val);
            head = head.next;
        }
    }

    public static ListNode mergeKLists(ListNode[] list) {
        // delete null node
        ListNode[] noEmpty=new ListNode[list.length];
        int k=0;
        for(int i=0;i<list.length;i++){
            if(list[i]!=null)noEmpty[k++]=list[i];
        }
        if(k==0)return null;
        if(k==1)return noEmpty[0];

        ListNode head = null;
        List<Integer> val = new ArrayList<Integer>();
        int i = 0;
        buildHead(noEmpty,k);
        print("build heap:", list,k);
        head = getOneNode(noEmpty, val,k);
        ListNode temp = head;
        while (temp != null) {
            temp.next = getOneNode(noEmpty, val,k);
            temp = temp.next;
        }

        // restore
        temp = head;
        while (temp != null) {
            if (temp.val == Integer.MAX_VALUE)
                temp.val = val.get(i++);
            temp = temp.next;
        }
        return head;
    }

    // get one node from heap,the first is the minValue.
    public static ListNode getOneNode(ListNode[] list, List<Integer> val,int k) {
        ListNode node = list[0];
        if (node.val == Integer.MAX_VALUE)
            return null;
        if (node.next != null) {
            list[0] = node.next;
        } else {
            val.add(list[0].val);
            list[0].val = Integer.MAX_VALUE;
        }
        adjustHeap(list, 0,k);
        print("AftergetOne:", list,k);
        return node;

    }

    public static void adjustHeap(ListNode[] list, int start,int k) {
        int left = 0, right = 0;
        int min = start;
        left = start * 2 + 1;
        right = start * 2 + 2;
        if(left<k&&list[left].val<list[min].val){
            min=left;
        }
        if(right<k&&list[right].val<list[min].val){
            min=right;
        }
        if (min!=start) {
            ListNode temp = list[min];
            list[min] = list[start];
            list[start] = temp;
            adjustHeap(list, min,k);
        }
    }

    // build a minHeadp
    public static void buildHead(ListNode[] list,int k) {
        for (int i = k/2; i >=0; i--) {
            adjustHeap(list, i,k);
        }

    }

    public static void print(String str, ListNode[] list,int k) {
        System.out.print(str);
        for (int i = 0; i < k; i++) {
            System.out.print(list[i].val + " ");
        }
        System.out.println();
    }
}

2、利用优先队列,非常简单。

public static ListNode mergeKLists(ListNode[] lists) {
        Queue<ListNode> queue = new PriorityQueue<ListNode>(
                new Comparator<ListNode>() {

                    @Override
                    public int compare(ListNode arg0, ListNode arg1) {
                        return arg0.val - arg1.val;
                    }

                });

        for (ListNode node : lists) {
            if (node != null) {
                queue.add(node);
            }
        }
        ListNode head = new ListNode(0), p = head, cur = null;
        while( !queue.isEmpty()) {
            cur = queue.poll();
            if(cur.next!=null)queue.offer(cur.next);
            p.next = cur;
            p = p.next;
        }
        return head.next;
    }

3、递归,每次将数组二分,求出二分后每一块的ListNode,然后合并这两块。

public static ListNode mergeKLists(ListNode[] lists){
    return partion(lists,0,lists.length-1);
}

public static ListNode partion(ListNode[] lists,int s,int e){
    if(s==e)  return lists[s];
    if(s<e){
        int q=s+(e-s)/2;
        ListNode l1=partion(lists,s,q);
        ListNode l2=partion(lists,q+1,e);
        return merge(l1,l2);
    }else
        return null;
}

//This function is from Merge Two Sorted Lists.
public static ListNode merge(ListNode l1,ListNode l2){
    if(l1==null) return l2;
    if(l2==null) return l1;
    if(l1.val<l2.val){
        l1.next=merge(l1.next,l2);
        return l1;
    }else{
        l2.next=merge(l1,l2.next);
        return l2;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值