leetcode——刪除鏈表的倒數第N個節點

這道題屬於學完數據結構的進階題，考研的時候經常會見到，現在leetcode見到了它

描述：

給定一個鏈表，刪除鏈表的倒數第 n 個節點，並且返回鏈表的頭結點。

示例：

給定一個鏈表: 1->2->3->4->5, 和 n = 2.

當刪除了倒數第二個節點後，鏈表變爲 1->2->3->5.
說明：

給定的 n 保證是有效的。

思路：

快慢兩個指針，快指針先走n步，之後一起走，當快指針走到鏈表尾的時候，慢指針下一個位置正好是倒數第n個結點

代碼：

ListNode* removeNthFromEnd(ListNode* head, int n) {
    if(!head || !head->next)
        return NULL;
    ListNode *slow = head, *fast = head;
    for(int i = 0; i < n; i++){
        fast = fast->next;
    }
    //刪除第一個結點的情況
    if(!fast){
        return head->next;
    }
    while(fast->next){
        fast = fast->next;
        slow = slow->next;
    }
    slow->next = slow->next->next;
    return head;
}