剑指 Offer II 021. 删除链表的倒数第 n 个结点

剑指 Offer II 021. 删除链表的倒数第 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

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

解法

暴力, 两次遍历

• 第一次遍历, 记录当前链表结点数量
• 第二次遍历, 使被删除结点的前一个结点指向被删除结点的后一个结点

代码

class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        int count = 0;
        ListNode* cur = head;
        while(cur != nullptr) {
            count ++;
            cur = cur->next;
        }
        if (count == n) return head->next;
        cur = head;
        count = count - n;
        while(--count) {
            cur = cur->next;
        }
        if (cur->next->next)
            cur->next = cur->next->next;
        else cur->next = nullptr;
        return head;
    }
};

复杂度分析

• 时间复杂度: O(n) n为链表长度
• 空间复杂度: O(1)

左右指针

前后指针相距n个结点, 当右指针到达链表尾, 左指针到达要删除结点的前一个

代码

class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        ListNode* left = head, *right = head;
        while(n --) {
            if (right->next) right = right->next;
            else return head->next; // 去掉删除头的情况
        }
        while(right->next != nullptr) {
            left = left->next;
            right = right->next;
        }
        left->next = left->next->next;
        return head;
    }
};

复杂度分析

• 时间复杂度: O(n) n为链表长度
• 空间复杂度: O(1)