力扣 21. 合并两个有序链表

57 阅读1分钟

image.png

image.png

起初我是用list1的头节点指向list2的头节点,但是这样做有问题,在处理不同的元素时要不停地保存list1和list2的下一个节点,且容易出错。看完官方的题解,我使用一个新建立的头节点pre,然后用一个tmp结点指向pre,tmp用来实际遍历,用tmp指向list1和list2中较小的那一个结点,思路就清晰多了,等最后返回结果时,还要返回结点pre.next。
/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode mergeTwoLists(ListNode list1, ListNode list2) {
        if(list1==null){
            return list2;
        } else if(list2==null){
            return list1;
        }
        ListNode pre=new ListNode(-1);
        ListNode tmp=pre;
        while(list1!=null&&list2!=null){
            if(list1.val<=list2.val){
                tmp.next=list1;
                list1=list1.next;
            } else {
                tmp.next=list2;
                list2=list2.next;
            }
            tmp=tmp.next;
        }
        if(list1==null)tmp.next=list2;
        if(list2==null)tmp.next=list1;
        return pre.next;
    }
}