[LeetCode19. 删除链表的倒数第 N 个结点] | 刷题打卡

183 阅读1分钟

一、题目描述:

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

进阶:你能尝试使用一趟扫描实现吗?

示例 1:

image.png

输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]

示例 2:

输入:head = [1], n = 1
输出:[]

示例 3:

输入:head = [1,2], n = 1
输出:[1]

二、思路分析:

删除链表的结点比较简单,但是这里要删除倒数的第N个,都知道题中的链表是单向的,这里想要删除倒数的第N个有两种思路,一种是利用栈,将链表全部入栈,然后出栈topN个,然后再把第N个删除。 时间复杂度O(n),空间复杂度O(n)

另一种是类似快慢走的解法,用双指针解决。可以想像成两个人走路,让一个人先走N步,第二个人再开始走,这样两人的距离就距离N,这是当前一个人到达末尾时,后一个人刚好是倒数第N的位置。这样就可以删除相应的结点了 时间复杂度O(n),空间复杂度O(1)

三、AC 代码:

class Solution {
    fun removeNthFromEnd(head: ListNode?, n: Int): ListNode? {
        val pre = ListNode(0)
        pre.next = head
        var p: ListNode? = pre
        var q: ListNode? = pre
        var count = n
        while (count > 0) {
            q = q?.next
            count -= 1
        }
        while (q?.next != null) {
            p = p?.next
            q = q.next
        }

        p?.next = p?.next?.next
        return pre.next
    }
}

四、总结:

经典的快慢指针问题。

本文正在参与「掘金 2021 春招闯关活动」, 点击查看活动详情