删除链表的倒数第 N 个元素

给定一个链表,删除链表的倒数第 个节点,并且返回链表的头结点。

示例:

给定一个链表: 1->2->3->4->5, 和 n = 2.

当删除了倒数第二个节点后,链表变为 1->2->3->5.

说明:

给定的 n 保证是有效的。

进阶:

你能尝试使用一趟扫描实现吗?

这道题最简单的做法是先遍历一次链表,得到长度,然后再利用for循环删除指定节点即可

最差的情况时间复杂度为O(2n)

比较快的方法是设立两个指针,先利用n使p指针往前移动一定的距离,然后再将p、q两个指针同时移动。

当p->next为空时,q的下一个节点就是要删除的指定节点了。

将输入处理一下,head为空啊,head->next为空啊这种情况排除掉,就可以了

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

class Solution(object):
    def removeNthFromEnd(self, head, n):
        """
        :type head: ListNode
        :type n: int
        :rtype: ListNode
        """
        if(n==None):return None #显然的None
        if(n==0):return head #不用动
        q = head #快指针
        p = head #慢指针
        for i in range(n): #快指针先动
            p = p.next
        if p == None: #此时需要注意的时,p == None 有两种情况,虽然都是把第一个节点删除,但q.next == None时,是因为head只有一个节点,q.next != None 时,就是在多个节点里删除第一个节点
            if q.next == None:
                return None
            else:
                head = head.next
                return head
        while(p.next!=None):#快慢指针一起动
            q = q.next
            p = p.next
        q.next = q.next.next#跳过指定节点
        return head

#因为比较简单,所以用的时间跟别人差不多

猜你喜欢

转载自blog.csdn.net/weixin_41169182/article/details/84946406