LeetCode 刷題日記 19. 刪除連結串列的倒數第N個節點

Alezzzzzz發表於2020-10-04

題目給出一個連結串列,給出一個整數n,要求返回刪除了倒數第n個節點的連結串列。這道題可以使用快慢指標法,讓fast先向後移n個節點,然後讓快慢指標同時向後移,直到快指標達到最後一個節點。此時,慢指標指向的下一個節點及為需要刪除的節點,將其刪除,程式碼如下:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        ListNode*  dummy = new ListNode(-1);
        dummy->next = head;  //虛擬頭節點
        ListNode *fast = dummy; //快指標
        ListNode *slow = dummy; //慢指標
        for(int i = n; i > 0; i--) //根據n,讓快指標先走n步
        {
           fast = fast -> next; 
        }  
        while(fast -> next) //當快指標還沒有到連結串列最後一個節點時,快慢指標同時向後移
        {
            fast = fast -> next;
            slow = slow -> next;
        }
        slow -> next = slow -> next -> next; //當快指標達到最後一個節點,slow所指向的節點的後一個節點為需要刪除節點,將其刪除
        return dummy->next;
    }
};

這道題易錯點在於,如果連結串列只有一個節點,n為1,此時,如果不採用虛擬頭節點,就會發生報錯。

相關文章