21.合并两个有序链表
将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例 1:
输入:l1 = [1,2,4], l2 = [1,3,4]
输出:[1,1,2,3,4,4]
示例 2:
输入:l1 = [], l2 = []
输出:[]
示例 3:
输入:l1 = [], l2 = [0]
输出:[0]
提示:
- 两个链表的节点数目范围是
[0, 50]
-100 <= Node.val <= 100
l1
和l2
均按 非递减顺序 排列
题解:
解法类似于归并排序中合并两个数组的部分,由于头节点不确定,因此需要创建一个dummy
节点,让p1
指向list1, p2
指向list2,比较p1和p2的值,较小的先放入到dummy所在链表中。
代码实现如下:
package com.offer;
import com.offer.leetcode.datastruct.ListNode;
import com.offer.leetcode.datastruct.ListNodeUtils;
public class _21合并两个有序链表 {
public static void main(String[] args) {
ListNode head1 = ListNodeUtils.createList(new int[]{1});
ListNode head2 = ListNodeUtils.createList(new int[]{1, 3, 4});
ListNode node = mergeTwoLists(head1, head2);
ListNodeUtils.printListNode(node);
}
public static ListNode mergeTwoLists(ListNode list1, ListNode list2) {
ListNode p1 = list1;
ListNode p2 = list2;
ListNode dummy = new ListNode(0);
ListNode pd = dummy.next;
ListNode pre = null;
while (p1 != null || p2 != null) {
pre = pd;
if (p1 != null && p2 != null) {
if (p1.val <= p2.val) {
pd = p1;
p1 = p1.next;
}else {
pd = p2;
p2 = p2.next;
}
}else if (p1 != null){
// p1不为空
pd = p1;
p1 = p1.next;
}else {
// p2不为空
pd = p2;
p2 = p2.next;
}
if (dummy.next == null) {
dummy.next = pd;
}else {
pre.next = pd;
}
}
return dummy.next;
}
}