提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
一看题目就有的惊才绝艳的思路(你好,我飘了(bushi))
给你一个链表,删除链表的倒数第 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
首先注意,此题有坑,题目中给的链表没有头结点,head是首节点。双指针思路如下:
令pq先指向首结点head,q指向首结点后的第n个结点,这样如果p和q同时后移,当q指向最后一个节点时,p指向倒数第n个结点的前一个结点,如此即可进行删除。
需要注意一些特殊情况:
①如果链表中只有一个元素,那么直接删掉此元素返回即可
②如果需要删掉链表中的第一个元素,那么q无法移动到首结点后的第n个结点(因为为空),所以如果出现这种移不够的情况,直接删掉首节点即可
# 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: Optional[ListNode], n: int) -> Optional[ListNode]:
#特殊情况
if(head.next==None):#只有一个元素
head=None
return head
#双指针
p=q=head
i=0
while(i<n and q.next != None): #q指向首结点后的第n个结点
q=q.next
i+=1
if(i<n): #i移动不完
head=head.next
return head
while(q.next != None): #p和q同时后移
p=p.next
q=q.next
p.next=p.next.next #删除
#返回
return head