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

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

解答

1. 双指针遍历一次链表，思路如下：

设置两个指针p，q，分别初始为头结点，然后p指针向后移动n次，接着p和q指针同时移动直到p指针指向链表的最后一个元素，此时q指针指向的是被删除元素的前一个元素，因此将需要被删除的元素进行删除，最终返回头结点即可，需要注意删除头结点的情况。实现代码如下：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        ListNode *q = head, *p = head, *t = head -> next;
        if(head == NULL)
        {
            return head;
        }
        while(n-- && p -> next != NULL) //p指针向后移动n位
        {
            p = p -> next;
        }
        if(n != -1) //删除头结点
        {
            t = head;
            head = head -> next;
            delete t;
            return head;
        }
        while(p -> next != NULL) //此时p和q指针一起向后移动，直到p指向最后一个结点
        {
            q = q -> next;
            p = p -> next;

        }
        t = q -> next;
        q -> next = t -> next;
        delete t;
        return head;
    }
};