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

https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/

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

  • 题目描述

  • 方法1: 快慢指针

    • 思路

    • 复杂度分析

    • 代码

题目描述

给你一个链表，删除链表的倒数第 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

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

方法1: 快慢指针

思路

找到倒数第 N 个节点

• 要删除倒数第 N 个节点，第一步要先找到倒数第 N 个节点(实际上对于单向链表，还需要找到倒数第 N+1 个节点才能完成删除倒数第 N 个节点的操作)。

• 简单的思路是，遍历确定链表长度 L，通过计算得到倒数第 N 个节点是顺数第 L-N 个节点。

• 快慢指针的思路是，定义两个指针，快指针先走 N 步，剩下的路程自然是 L-N，这时再让快慢指针同时走，等快指针走路末尾时，慢指针走过的路程刚好是 L-N，也就是刚好走到倒数第 N 个节点。

使用 dummy 节点来避免单独处理头节点

假如要删除的节点刚好是头节点的话，要额外处理就很麻烦。但如果事先在头节点前面添加一个 dummy 节点，就可以把头节点当成一个普通节点来处理了，这是链表题中常用的技巧。

复杂度分析

• 时间复杂度：$O(N)$。

• 空间复杂度：$O(1)$。

代码

C++ Code

/**
 * 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 *fast = head, *slow = head;
        while (n--) {
            fast = fast->next;
        }
        // 额外处理头部节点
        if (!fast) {
            ListNode *new_head = head->next;
            head->next = nullptr;
            return new_head;
        }
        while (fast->next) {
            fast = fast->next;
            slow = slow->next;
        }
        ListNode *target = slow->next;
        slow->next = target->next;
        target->next = nullptr;
        return head;
    }
};

使用 dummy 节点

/**
 * 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 *dummy = new ListNode(0, head);
        ListNode *fast = dummy, *slow = dummy;

        // 快指针先走 n 步
        while (n-- > 0) {
            fast = fast->next;
        }
        // 当快指针走到尾节点时
        // 慢指针刚好走到倒数 n+1 个节点
        while (fast && fast->next) {
            fast = fast->next;
            slow = slow->next;
        }
        ListNode *target = slow->next;
        slow->next = target->next;
        target->next = nullptr;
        return dummy->next;
    }
};

更多题解可以访问：https://github.com/suukii/91-days-algorithm