LeetCode19 移除倒数第N个元素

链接

Remove Nth Node From End of List


难度

Medium


描述

Given a linked list, remove the n -th node from the end of list and return
its head.

给定一个链表,要求移除导数第n个元素,并且返回新链表的head


样例:

Given linked list: **1- >2->3->4->5**, and **_n_ = 2**.

After removing the second node from the end, the linked list becomes **1- >2->3->5**.


注意:

Given n will always be valid.

扫描二维码关注公众号,回复: 9049557 查看本文章

给定的n永远有效


Follow up:

Could you do this in one pass?

你能一发通过么???官方嘲讽。


题解


题目的意思非常直白,不需要太多思考就能理解。但是上手去做的话会有一点小问题,因为如果是数组很好办,我们直接可以求到数组的长度,导数第N个元素也非常容易确定。但是如果是链表的话就不同了,因为我们并不知道链表的长度,所以没办法直接获取需要删除节点的位置。

既然直接没有办法求到,那么可以间接去求嘛,所以很自然地可以想到两次遍历的方法。


两次遍历


这个方法非常直观,基本上可以说是顾名思义。我们对这个链表遍历两次,第一次求到链表的长度,这样我们就可以推算到倒数第N个数是正数第几个数了。第二次我们移动对应的长度,找到需要删除的节点,将它移除即可。

我们来看下面这张图,完全可以脑补出算法:

虽然算法不难,但是这题当中藏着trick,隐藏了出题人深深的恶意。不相信的话,可以试着不看答案写写看,基本上一定会错上一两次。原因很简单,因为会有一些特殊情况需要考虑。

比如,特殊情况1:链表当中只有一个元素,显然这个时候根本不需要移动,也不用删除,直接return None就好了。但是如果我们使用常规方法的话,是无法删掉的,必须要特殊判断这种情况。

特殊情况2:这个要删的元素刚好是第一个head元素,这种情况也没有办法常规解决,也需要特殊判断。

把这两个特殊情况考虑到,基本上就没问题了。

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
        l = 0
        pnt = head
        # 计算链表长度
        while pnt:
            l += 1
            pnt = pnt.next
        # 如果长度为1,直接return None
        if l == 1:
            return None
        # 计算删除需要移动的长度
        l = l - n - 1
        # 如果小于0,说明需要删除第一个元素,那么直接return head.next。
        if l < 0:
            return head.next
        pnt = head
        for i in range(l):
            pnt = pnt.next
        pnt.next = pnt.next.next
        return head


一次遍历


上面的这种算法中规中矩,基本上没有难度。那么有没有更好的算法,比如我们能不能做到只遍历链表一次呢?这样不就优化了复杂度了吗?

当然是可以的,说来这种算法也非常巧妙。如果说我们把链表想象成一条跑道,而把移动遍历的指针想象成跑道上的运动员。我们现在不知道跑道有多长,但是我们想要知道距离终点30米的位置。我们还知道运动员的奔跑速度都一样。应该怎么做呢?

我们可以先让一个运动员先跑30米,然后再派另外一个运动员从起点出发。这样,当第一个运动员跑到终点的时候,第二个运动员所在的位置就是距离终点30米的位置。

我们把上述的运动员换成指针,跑道换成链表,就是这题的解法了。

同样,我们也有一张图可以说明:

我们根据描述,可以试着写出代码。

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
        pnt1 = head
        
        # 第一个指针先跑n步
        for i in range(n):
            pnt1 = pnt1.next
            # 注意,如果是删除head,会出现跑过头的情况,需要判断
            if pnt1 is None:
                return head.next
            
        pnt2 = head
        # 移动第一个指针,直到结尾
        while pnt1.next:
            pnt1 = pnt1.next
            pnt2 = pnt2.next
        
        # 删除pnt2.next位置的节点
        pnt2.next = pnt2.next.next
        return head

看起来第二种方法似乎更优一些,但实际上如果分析复杂度的话,两者都是\(O(n)\)的复杂度,并没有高下之分。而且比较奇葩的是,我用CPP提交的时候是能明显感觉出来第二种方法优于第一种几毫秒的,但是当我换成了Python之后,第二种方法的耗时反而还更长了。可见,快慢都是相对的,一般情况下来说在复杂度相同的情况下,优化常数意义不大。

虽然搞了半天,并没有什么效果,但至少我们明白了这点,也算是收获。

好了,今天的文章就是这些,如果觉得有所收获,请顺手扫码点个关注,你们的举手之劳对我来说很重要。

猜你喜欢

转载自www.cnblogs.com/techflow/p/12289784.html