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

Posted 2021-03-17 kennyoooo

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

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

难度	完成日期	耗时	提交次数
中等	2020-1-10	0.5小时	1

问题描述

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

示例：

给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后，链表变为 1->2->3->5.

解题思路

普通方法

ListNode *removeNthFromEnd(ListNode *head, int n) {
    ListNode *_head = head;
    int length = 1;
    while (head->next != nullptr) {
        length++;
        head = head->next;
    }
    if (n == length) {
        return _head->next;
    }
    head = _head;
    int position = length - n;
    for (int i = 1; i < position; i++) {
        head = head->next;
    }
    head->next = head->next->next;
    return _head;
}

先求出链表长度，再按顺序求出删除链表的实际位置，删除节点，连接到下一个节点。注意判断删除的是否为头节点。

尝试使用一趟扫描实现

ListNode *removeNthFromEnd(ListNode *head, int n) {
    ListNode *_head = head;
    ListNode *end = head;
    for (int i = 0; i < n; i++) {
        end = end->next;
    }
    if (end == nullptr) {
        return _head->next;
    }
    while (end->next != nullptr) {
        head = head->next;
        end = end->next;
    }
    head->next = head->next->next;
    return _head;
}

使用两个指针，分别保存当前遍历位置和 n 个节点后位置，若 n 个节点后已经为最后一个节点，则删除当前遍历位置节点。