想学习机器学习算法的小伙伴,欢迎关注我,专栏https://blog.csdn.net/qq_36583400/category_11959999.html
本文隶属系列专栏https://blog.csdn.net/qq_36583400/category_12074252.html
一起学习,一起进步,做时代有为青年~
本节想穿插着讨论一下链表相关的一道题。
题目来自于LeetCode第19题力扣https://leetcode.cn/problems/remove-nth-node-from-end-of-list/
问题描述:
解题思路:
这是一道中等题,如果我们使用遍历两次的方式,那这道题就很简单了,先遍历一次,获得链表长度,再遍历一次直到要删除的节点位置的前一个位置,然后删除要删除的节点即可。那么如果只用一次遍历该怎么解呢?
我们知道,倒数第N,可以用正数第N表示,如下图:
假如我们要删除倒数第3个节点,为了只用一次遍历就找到倒数第3个节点,我们可以先找到正数第3个节点,然后首节点指针和第3个节点指针同时往后走,当第二个指针走到尾节点位置,那么第一个指针指向的节点,就是我们要删除的节点。这是一方面,另一方面,我们题目要的是删除节点,在链表中删除节点,通常的做法就是让要删除节点的前一个节点直接指向删除节点的后一个节点,所以这里我们的第一第二个指针的走法需要改变一下,我们让后一个指针再往后错一位,这样一来,当第二个指针走到尾节点时,第一个指针指向的就是删除节点的前一个节点了,如下图:
实现上,首先定义快慢两个指针,让快指针先走,走到与慢指针间隔为N的地方(N为题目给出的倒数第N), 然后两个指针同时走,虽然两个指针同时走,但是还是需要一前一后跟着走,大家可以用手比划一下,就可以知道,应该让快指针总是比慢指针先行一步,这样最后慢指针才能到达要求位置。注意考虑边界条件:当要删除的节点为第一个节点,这时候由于第一个节点前方没有节点,所以当快指针移动到间隔慢指针为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 not head.next and n==1:
return None
dummy = ListNode(None)
dummy.next = head
slow = fast = head
while n:
fast = fast.next
n -= 1
if not fast:
return dummy.next.next
while fast.next:
fast = fast.next
slow = slow.next
temp = slow.next.next
slow.next = temp
return dummy.next