给你一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
示例 2:
输入:head = [1], n = 1
输出:[]
示例 3:
输入:head = [1,2], n = 1
输出:[1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
代码:
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
low = fast = head
for i in range(n):
fast = fast.next
if not fast:
return head.next
while fast.next:
low, fast = low.next, fast.next
low.next = low.next.next
return head