【Leetcode】【python】Remove Nth Node From End of List

题目大意

删除链表中倒数第n个节点

解题思路

加一个虚假头结点dummy,并使用双指针p1和p2。p1先向前移动n个节点(从dummy节点开始移动,所以移动了n其实是移动到了前一位),然后p1和p2同时移动,当p1.next==None时,此时p2.next指的就是需要删除的节点前面一个节点,将其指向.next.next即可

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
class Solution(object):
def removeNthFromEnd(self, head, n):
"""
:type head: ListNode
:type n: int
:rtype: ListNode
"""
dummy = ListNode(0)
dummy.next = head
p1 = p2 = dummy
while n:
p1=p1.next
n -= 1
while p1.next: # p1后指针指向空后跳出
p1 = p1.next
p2 = p2.next
p2.next = p2.next.next
return dummy.next

总结