给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
最先想到的做法一定是先找出链表的长度,然后正向找到倒数第N个节点。这样思路简单,但是效率估计就没有那么高效了。
先展示一下我用第一种做法写出的代码:
class Solution(object):
def removeNthFromEnd(self, head, n):
"""
:type head: ListNode
:type n: int
:rtype: ListNode
"""
p=q=head
s=0
if p==None :
return head
while p:
s+=1
p=p.next
s=s-n
if s==0:
head=head.next
return head
while s>1:
s-=1
q=q.next
q.next=q.next.next
return head
更改一下思路,设立两个快慢指针,快指针先走n步,之后设立循环,在快指针指到空节点之前,快慢指针一起走,这样慢指针在循环完之后会留下n个节点,也就得到了倒数第N个节点。可是,这里需要注意一个问题,如果,直接快慢指针走下去的话,会出现越界的问题。比如,链表长度为2,要删除倒数第N个节点,按照上面的思路,走n步后,快指针应该指在最后一个节点,那在循环时,我们的判断语句肯定是:
while l1.next:
看代码:
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
l1=l2=head
if l1==None:
return head
if l1.next==None and n==1:
return None
for i in range(n):
l1=l1.next
while l1.next:
l1=l1.next
l2=l2.next
l2.next=l2.next.next
return head
这样l1本来就在最后一个节点,哪里还有next。
所以有一个解决办法,添加一个头节点:
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
head2=ListNode(0)
head2.next=head
l1=l2=head2
if l1.next==None or (l1.next==None and n==1):
return None
for i in range(n):
l1=l1.next
while l1.next:
l1=l1.next
l2=l2.next
l2.next=l2.next.next
return head2.next
这样就可以解决了。