链表练习(四)— 合并K个升序链表

题目:
给定K个升序链表,将链表合并,并返回合并后链表的head节点(主要用到了上期博客中提到的PriorityQueue和比较器,不了解的可以去看下)。

分析:
可将K个节点的head都放入PriorityQueue中并指定排序方法,小的弹出,并将下一节点压入 关于PriorityQueue

  1. 遍历Node数组,将每一个链表的head放入PriorityQueue中。
  2. PriorityQueue.poll()方法,弹出PriorityQueue中最小的Node节点作为新链表的head。
  3. 将最小head节点的next节点压入PriorityQueue中,并声明变量pre = head,用来链接head的下一个节点。
  4. 循环,如果PriorityQueue不为空。
  5. 再次弹出PriorityQueue中最小节点cur,获取,并将pre的next指针指向cur,pre = cur(获取到当前新链表的节点,目的下次找到最小节点的指向)。
  6. 如果当前弹出Node的next不为null,则将下一节点压入PriorityQueue

完整代码

	 public static class Node {
	        public int val;
	        public Node next;
	
	        public Node(int value) {
	            this.val = value;
	        }
	    }
	 //根据Node的val属性进行排序
	 public static class NodeComparator implements Comparator<Node> {
	
	        @Override
	        public int compare(Node o1, Node o2) {
	            return o1.val - o2.val;
	        }
	    }
    
	public static Node mergeNode(Node[] nodes) {
	        if (nodes == null) {
	            return null;
	        }
			//创建优先级队列,并指定排序规则
	        PriorityQueue<Node> priorityQueue = new PriorityQueue<>(new NodeComparator());
			//将每个链表的head节点加入到队列中
	        for (int i = 0; i < nodes.length; i++) {
	            if (nodes[i] != null) {
	                priorityQueue.add(nodes[i]);
	            }
	        }
			if (priorityQueue.isEmpty()) {
						return null;
					}
			//弹出当前队列中最小节点,作为新链表的head节点
	        Node head = priorityQueue.poll();
	        //声明pre变量指向head,目的是找到新的最小节点后,head的next能够指向
	        Node pre = head;
	        //如果当前head的next不为null,则压入队列
	        if (pre.next != null) {
	            priorityQueue.add(pre.next);
	        }
			//优先级队列不为空则一直遍历
	        while (!priorityQueue.isEmpty()) {
	        	//弹出当前优先级队列中最小节点
	            Node cur = priorityQueue.poll();
	            //新head的next指针指向当前节点
	            pre.next = cur;
	            //记住当前节点,也是便于接下来的next指向
	            pre = cur;
	            //如果节点不为null,则继续压入priorityQueue
	            if (cur.next != null) {
	                priorityQueue.add(cur.next);
	            }
	        }
	        return head;
	    }
    //测试
	public static void main(String[] args) {
	        Node node1 = new Node(1);
	        node1.next = new Node(5);
	        node1.next.next = new Node(7);
	
	
	        Node node2 = new Node(1);
	        node2.next = new Node(4);
	        node2.next.next = new Node(8);
	
	        Node node3 = new Node(5);
	        node3.next = new Node(9);
	        node3.next.next = new Node(9);
	
	        Node[] nodes = {node1,node2,node3};
	        Node node = mergeNode(nodes);
	
	        while (node != null){
	            System.out.print(node.val + " ");
	            node = node.next;
	        }
	    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要实现合并K个升序链表,我们可以考虑使用分治法,将K个链表划分为两个子问题,分别合并这两个子问题,然后不断递归下去。 具体实现过程如下: 1. 将K个链表按照长度平均划分为两个子问题,每个子问题递归调用合并函数,直到只剩下一个链表或两个链表。 2. 合并两个链表的过程可以使用归并排序中的合并函数,将两个链表合并为一个升序链表。 3. 将合并后的链表返回,然后递归回去继续合并两个子问题的结果。 Java代码实现如下: ``` public ListNode mergeKLists(ListNode[] lists) { if (lists == null || lists.length == 0) { return null; } return mergeKLists(lists, 0, lists.length - 1); } private ListNode mergeKLists(ListNode[] lists, int left, int right) { if (left == right) { return lists[left]; } int mid = (left + right) / 2; ListNode l1 = mergeKLists(lists, left, mid); ListNode l2 = mergeKLists(lists, mid + 1, right); return mergeTwoLists(l1, l2); } private ListNode mergeTwoLists(ListNode l1, ListNode l2) { if (l1 == null) { return l2; } if (l2 == null) { return l1; } if (l1.val < l2.val) { l1.next = mergeTwoLists(l1.next, l2); return l1; } else { l2.next = mergeTwoLists(l1, l2.next); return l2; } } ``` 其中,`mergeKLists` 函数是递归调用的入口函数,它接收一个 `ListNode` 数组作为参数,表示要合并的K个链表。在函数中,我们首先判断链表数组是否为空或长度为0,如果是,则返回 `null`。否则,我们调用 `mergeKLists` 函数,将链表数组划分为两个子问题,然后递归调用 `mergeKLists` 函数,继续划分子问题,直到只剩下一个链表或两个链表。 在 `mergeKLists` 函数中,我们使用归并排序的思想,将两个链表合并为一个升序链表。具体实现是在 `mergeTwoLists` 函数中,它接收两个链表作为参数,递归调用自身,将两个链表合并为一个升序链表
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值