資料結構與演算法 | Leetcode 21:Merge Two Sorted Lists

wangwei_hz發表於2019-01-08

bicycle_journey-wallpaper

原文連結:wangwei.one/posts/java-…

前面,我們實現了連結串列的 環檢測 操作,本篇來聊聊,如何合併兩個有序連結串列。

有序連結串列合併

Leetcode 21:Merge Two Sorted Lists

示例

Input: 1->2->4, 1->3->4
Output: 1->1->2->3->4->4
複製程式碼

使用虛假的Head節點

定義一個臨時虛假的Head節點,再建立一個指向tail的指標,以便於在尾部新增節點。

對ListNode1和ListNode2同時進行遍歷,比較每次取出來的節點大小,並繫結到前面tail指標上去,直到最終所有的元素全部遍歷完。

最後,返回 dummyNode.next ,即為新連結串列的head節點。

程式碼

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    
    public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        ListNode dummyNode = new ListNode(0);
        ListNode tail = dummyNode;  
    
        while(true){
            
            if(l1 == null){
                tail.next = l2;
                break;
            }
            
            if(l2 == null){
                tail.next = l1;
                break;
            }
    
            ListNode next1 = l1;
            ListNode next2 = l2;
            
            if(next1.val <= next2.val){
                tail.next = next1;
                l1 = l1.next;
            }else{
                tail.next = next2;
                l2 = l2.next;
            }
            
            tail = tail.next;
        }
        
        return dummyNode.next;
        
    }
        
}
複製程式碼

遞迴

使用遞迴的方式,程式碼比遍歷看上去簡潔很多,但是它所佔用的棧空間會隨著連結串列節點數量的增加而增加。

程式碼

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    
    public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        
        ListNode result = null;
        
        if(l1 == null){
            return l2;
        }
        if(l2 == null){
            return l1;
        }
        
        if(l1.val <= l2.val){
            result = l1;
            result.next = mergeTwoLists(l1.next, l2);
        }else{
            result = l2;
            result.next = mergeTwoLists(l1, l2.next);
        }
        return result;
    }
    
}
複製程式碼

相關文章