给你一个链表,删除链表的倒数第 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
一解
使用双指针 first 和 second :
- 初始化时,first 比 second 快 n 个结点。
- first 和 second 同步移动,当 first 到结点尾时, second 指向要删除结点的前一个结点。
- 利用 second 删除目标结点。
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
# 使用额外节点,方便处理只有一个结点,或者 n 指向头结点的情况
dump = ListNode(0, head)
first, second = head, dump
for _ in range(n):
first = first.next
while first:
first = first.next
second = second.next
second.next = second.next.next
return dump.next
- 时间复杂度 O(n):其中 n 为结点长度。
- 空间复杂度 O(1)。