进击的小白Day032——LeetCode刷题笔记010：删除链表的倒数第N个节点

题目来自LeetCode。
题目描述：
给定一个链表，删除链表的倒数第 n 个节点，并且返回链表的头结点。
示例：
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后，链表变为 1->2->3->5.
说明：
给定的 n 保证是有效的。
进阶：
你能尝试使用一趟扫描实现吗？

单指针遍历两次：

struct ListNode* removeNthFromEnd(struct ListNode* head, int n) {
    struct ListNode *p = head;
    int num = 0, m = 0, i = 1;
    
    while(p != NULL)
    {
        num++;
        p = p->next;
    }
    
    if (num == 0 || (num == 1 && n == 1))
    {
        head = NULL;
        return head;
    }
    else if (num == 1 && n == 0)
    {
        return head;
    }
    
    m = num + 1 - n;
    if (m == 1)
    {
        head = head->next;
        return head;
    }
    
    p = head;
    
    while(i < m - 1)
    {
        p = p->next;
        i++;
    }
    
    p->next = p->next->next;
    
    return head;
}

借鉴了别人的代码，优化了一下，双指针，遍历一次即可定位到要删除的节点：

struct ListNode* removeNthFromEnd(struct ListNode* head, int n) {
    struct ListNode *p = head;
    struct ListNode *q = head;
    int i = 0;
    
    for (i = 0; i < n; i++)
    {
        p = p->next;
    }
    if (p == NULL)
    {
        head = head->next;
        return head;
    }
    while (p->next != NULL)
    {
        p = p->next;
        q = q->next;
    }
    
    q->next = q->next->next;
    return head;
}