刷题-力扣-19

19. 删除链表的倒数第 N 个结点

题目链接

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目描述

给你一个链表，删除链表的倒数第 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. 根据题目描述删除倒数第n个元素，首先计算链表长度
2. 获取待删除的元素的前一个位置
3. 删除指定位置元素
4. 注意判断特殊情况。如待删除元素没有后继、待删除为链表头元素

代码

/**
 * 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 *p = head;
        int len = 0;
        while (p) {
            len++;
            p = p->next;
        }
        p = head;
        if (len - n == 0) {
            head = head->next;
            return head;
        }
        for (int i = 1; i < len - n; i++)  p = p->next;
        if (p->next) {
            if (p->next->next) p->next = p->next->next;
            else p->next = nullptr;
        } else head = nullptr;
        return head;
    }
};

用时26min