一、题目描述:
给你一个链表,删除链表的倒数第 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]
二、思路分析:
删除链表的结点比较简单,但是这里要删除倒数的第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 春招闯关活动」, 点击查看活动详情