算法通关村第一关—链表经典问题之合并两个有序链表笔记
源码地址:GitHub-算法通关村
题目地址:LeetCode
解题思路:
- 创建一个虚拟头节点
preHead
作为合并后链表的起始点,并创建一个当前节点指针preNode
初始化指向preHead
。 - 使用 while 循环,遍历两个有序链表
list1
和list2
,直到其中一个链表为空。 - 在循环中,比较
list1
和list2
当前节点的值,将较小的节点接入合并链表,并将相应的链表指针向后移动。 - 更新
preNode
指向新接入的节点,使其指向合并链表的最后一个节点。 - 循环结束后,其中一个链表已经遍历完毕,可能还有一个链表中的节点未处理完毕。此时,直接将剩余未处理的链表部分接入合并链表的末尾。
- 返回
preHead.next
,即合并后的有序链表的起始节点。
/**
* 虚拟头结点和双指针
*
* @param list1
* @param list2
* @return
*/
public ListNode mergeTwoLists(ListNode list1, ListNode list2) {
ListNode preHead = new ListNode(-1);
ListNode preNode = preHead;
while (list1 != null && list2 != null) {
if (list1.val <= list2.val) {
preNode.next = list1;
list1 = list1.next;
} else {
preNode.next = list2;
list2 = list2.next;
}
preNode = preNode.next;
}
preNode.next = list1 == null ? list2 : list1;
return preHead.next;
}
变型
/**
* 合并K个链表
*
* @param lists
* @return
*/
public ListNode mergeKLists(ListNode[] lists) {
ListNode res = null;
for (ListNode list : lists) {
res = mergeTwoLists(res, list);
}
return res;
}