前面,我們實現了連結串列的 環檢測 操作,本篇來聊聊,如何合併兩個有序連結串列。
有序連結串列合併
示例
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;
}
}
複製程式碼