前面,我們實現了 兩個有序連結串列的合併 操作,本篇來聊聊,如何刪除一個連結串列的倒數第N個節點。
刪除單連結串列倒數第N個節點
給定一個單連結串列,如: 1->2->3->4->5
,要求刪除倒數第N個節點,假設 N = 2
,並返回頭節點。
則返回結果:1->2->3->5
.
解法一
這一題的難度標記為 medium
,解法一比較容易想出來,我個人覺得難度不大。
思路
迴圈兩遍:
- 先遍歷一遍,求得整個連結串列的長度。
- 再遍歷一遍,當總長度
len
減去n
,恰好等於迴圈的下標i
時,就找到對應要刪除的目標元素,將prev
節點與next
節點連線起來即可。
程式碼
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
if(head == null){
return null;
}
int len = 0;
for(ListNode curr = head ; curr != null;){
len++;
curr = curr.next;
}
if(len == 0){
return null;
}
// remove head
if(len == n){
return head.next;
}
ListNode prev = null;
int i = 0;
for(ListNode curr = head; curr != null;){
i++;
prev = curr;
curr = curr.next;
if(i == (len - n)){
prev.next = curr.next;
}
}
return head;
}
}
複製程式碼
Leetcode測試的執行時間為6ms
,超過了98.75%
的java程式碼。
解法二
這種解法,比較巧妙,沒有想出來,查了網上的解法,思路如下:
思路
只需要迴圈一遍,定義兩個指標,一個快指標,一個慢指標,讓快指標的巧好領先於慢指標n
步。當快指標到達tail節點時,滿指標巧好就是我們需要刪除的目標元素。
程式碼
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
if(head == null){
return null;
}
ListNode fast = head;
ListNode slow = head;
for(int i = 0; i < n; i++){
fast = fast.next;
}
if(fast == null){
return slow.next;
}
ListNode prev = null;
for(ListNode curr = slow; curr != null; ){
// when fast arrived at tail, remove slow.
if(fast == null){
prev.next = curr.next;
break;
}
prev = curr;
curr = curr.next;
// move fast forward
fast = fast.next;
}
return head;
}
}
複製程式碼
這段程式碼在LeetCode上的測試結果與解法一的一樣。
這種解法與之前的 連結串列環檢測 題目中都使用到了快慢指標,用來定位特定的元素。