LeetCode ---- 删除链表的倒数第N个节点

最新推荐文章于 2022-04-25 19:58:32 发布

袁小黑

最新推荐文章于 2022-04-25 19:58:32 发布

阅读量229

点赞数

CC 4.0 BY-SA版权

分类专栏：题目

本文链接：https://blog.youkuaiyun.com/ydonghao2/article/details/80280815

题目专栏收录该内容

52 篇文章

订阅专栏

本文介绍了一种算法，该算法可以实现在单次遍历中删除给定链表的倒数第N个节点，并返回修改后的链表头结点。通过使用栈来追踪节点，确保操作的准确性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

示例：

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

说明：

给定的 n 保证是有效的。

进阶：

你能尝试使用一趟扫描实现吗？

public class RemoveNthFromEnd {

    @Test
    public void removeNthFromEndTest() {
        Assert.assertNull(removeNthFromEnd(new ListNode(1), 1));
        ListNode listNode = new ListNode(1);
        listNode.next = new ListNode(2);
        Assert.assertEquals(2, removeNthFromEnd(listNode, 2).val);
    }

    @Test
    public void removeNthFromEndTest1() {
        ListNode listNode = new ListNode(1);
        listNode.next = new ListNode(2);
        Assert.assertEquals(1, removeNthFromEnd(listNode, 1).val);
    }
    public ListNode removeNthFromEnd(ListNode head, int n) {
        ListNode temp = head;
        Stack<ListNode> stack = new Stack<>();
        while (head != null) {
            stack.push(head);
            head = head.next;
        }

        for (int i = 0; i < n-1; i++) {
            stack.pop();
        }

        ListNode targetNode = stack.pop();
        if (stack.size() == 0) {
            temp = targetNode.next;
        } else if(n == 1) {
            ListNode lastNode = stack.pop();
            lastNode.next = null;
        } else {
            targetNode.val = targetNode.next.val;
            targetNode.next = targetNode.next.next;
        }
        return temp;
    }

}