leetcode刷题(剑指offer) 21.合并两个有序链表

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;
    }
}