資料結構與演算法 | Leetcode 19. Remove Nth Node From End of List

wangwei_hz發表於2019-01-24

puppy-dog-cute-love-cat-mammal

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

前面,我們實現了 兩個有序連結串列的合併 操作,本篇來聊聊,如何刪除一個連結串列的倒數第N個節點。

刪除單連結串列倒數第N個節點

Leetcode 19. Remove Nth Node From End of List

給定一個單連結串列,如: 1->2->3->4->5,要求刪除倒數第N個節點,假設 N = 2,並返回頭節點。

則返回結果:1->2->3->5 .

解法一

這一題的難度標記為 medium,解法一比較容易想出來,我個人覺得難度不大。

思路

迴圈兩遍:

  1. 先遍歷一遍,求得整個連結串列的長度。
  2. 再遍歷一遍,當總長度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上的測試結果與解法一的一樣。

這種解法與之前的 連結串列環檢測 題目中都使用到了快慢指標,用來定位特定的元素。

相關練習

參考資料

相關文章