LeetCode 19. 删除链表的倒数第N个节点(C++)

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

示例：

给定一个链表: 1->2->3->4->5, 和 n = 2.

当删除了倒数第二个节点后，链表变为 1->2->3->5.

说明：

给定的 n 保证是有效的。

进阶：

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

思路：双指针，控制两个指针距离为n，同时后移，然后删除第一个指针的下一个元素，注意不要访问空指针，比如p->next,p->next->next都不能为空。

我的解答：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
static const auto _ = []() {
    ios::sync_with_stdio(false); 
    cin.tie(nullptr); 
    return nullptr; 
}();
class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        if(!(head->next))return NULL;
        ListNode* p=head, * q=head;
        int i=0;
        while(i<n&&q->next)
        {
            q=q->next;
            i++;
        }
        if(i<n)
            return head->next;
        while(q->next)
        {
            p=p->next;
            q=q->next;
        }
        if(p->next->next)
            p->next=p->next->next;
        else
            p->next=NULL;
        return head;
    }
};