leetcode 19-删除链表的倒数第 n 个结点

给你一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点。

示例 1：

输入：head = [1,2,3,4,5], n = 2
输出：[1,2,3,5]

示例 2：

输入：head = [1], n = 1
输出：[]

示例 3：

输入：head = [1,2], n = 1
输出：[1]

提示：

链表中结点的数目为 sz
1 <= sz <= 30
0 <= Node.val <= 100
1 <= n <= sz

进阶：你能尝试使用一趟扫描实现吗？

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
 
/*
先计算出链表的长度；
若删除的是第一个节点，则直接返回head->naxt,因为这个链表是不带头节点的链表，不能用p->next = p->next->next，原因是若该链表只有一个节点则不存在p->next->next；
若不是第一个节点则可以找到倒数n+1个节点，然后删除其后紧邻的节点。   
*/
struct ListNode* removeNthFromEnd(struct ListNode* head, int n){
    struct ListNode *p = head;
    int num = 0;
    while(p){
        num++;
        p = p->next;
    }
    if(num - n == 0)
        return head->next;
    p = head;
    for( ; num - 1 > n; num--)
        p = p->next;

    p->next = p->next->next;
    return head;
}