Loading

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

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

作者:Grey

原文地址:

博客园:删除链表的倒数第 N 个结点

CSDN:删除链表的倒数第 N 个结点

题目描述

题目链接

思路

由于题目要求: Could you do this in one pass?

所以这题的难点在于:不通过获取链表长度的方式来拿需要删除的节点。

要定位到需要删除的点,可以这样操作,假设我们要删除的点是X,会有如下关系

image

其中:

  • X'点是 X 关于中点的对称点

  • Y 是 X 前一个点

  • Y' 是 X' 后一个点

所以,可以很容易的得到一个结论:

如果设置两个指针,假设一个叫 pre,一个叫 cur,

  1. cur 从头开始走,当 cur 走到 Y' 位置的时候(即走了 n + 1 步),pre 此时开始从头走;

  2. 接下来 cur 和 pre 同时向下一个位置移动,当 cur 移动到结尾(cur == null)的时候,pre 应该正好在 Y 位置;

  3. 然后调用:pre.next = pre.next.next;

这样就把需要删除的 X 位置给删除掉了。

这道题还有两个特别需要注意的点是:

  • 需要被删除的节点如果不是头节点,那么返回的链表节点应该是头节点。

  • 如果需要删除的节点正好是头节点,那么返回的节点应该是头节点的下一个节点。

所以,如果上述三个步骤中的第一步:

cur 从头开始走,当 cur 走到 Y' 位置的时候(即走了 n + 1 步),正好是 cur == null 的位置。那 pre 不用走了,头节点即为需要删除的节点。

直接返回 head.next 即可。

完整代码:

    public static ListNode removeNthFromEnd(ListNode head, int n) {
        ListNode cur = head;
        ListNode pre = null;
        while (cur != null) {
            n--;
            if (n == -1) {
                pre = head;
            }
            if (n < -1) {
                pre = pre.next;
            }
            cur = cur.next;
        } 
        if (pre == null) {
            return head.next;
        }
        pre.next = pre.next.next;
        return head;
    }

更多

算法和数据结构笔记

posted @ 2021-04-19 00:16  Grey Zeng  阅读(93)  评论(0编辑  收藏  举报