该题目和以下两个题的解题思路相似:
目录
一、题目
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
二、示例
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
- 给定的 n 保证是有效的。
进阶:
- 你能尝试使用一趟扫描实现吗?
三、思路
双指针思想,p,q两指针。
p指针是快指针,q指针是慢指针。
使两指针的距离为n,那么,当快指针p遍历到链表结尾时,慢指针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: ListNode, n: int):
p = head
q = head
for i in range(n):
if p.next:
p = p.next
else:
return head.next
while p.next:
p = p.next
q = q.next
q.next = q.next.next
return head
if __name__ == '__main__':
head = ListNode(1)
head.next = ListNode(2)
head.next.next = ListNode(3)
head.next.next.next = ListNode(4)
head.next.next.next.next = ListNode(5)
n = 2
s = Solution()
new_head = s.removeNthFromEnd(head, n)
print(new_head)