菜鸟刷题——19删除链表的倒数第N个结点

题目描述

这个题目属于链表的常规方法，最常能想到的就是快慢指针法，由于本人对于递归的方法运用不熟练，因此下面介绍两种方法来解决这个问题。

方法一：快慢指针法

定义两个指针current,last.使得last指针先走n-1步，然后再current和last一起走，这样当last.next==None时，current指针指向的下一个结点就是需要删除的结点。这里要注意，当链表元素只有1个且需要删除1个元素的时，会导致当前结点无法删除的现象。因此我们使用一个结点p,使得p.next=head,current=dummy, last=head开始进行上述操作，完整代码如下：

'''给你一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点。 '''
class ListNode(object):
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next
class LinkList(object):
    def __init__(self):
        self.head = None
    def linklist(self, data):
        if not data: return
        dummy = ListNode(-1)
        p = dummy
        p.next = ListNode(data[0])
        p = p.next
        for i in data[1:]:
            p.next =ListNode(i)
            p = p.next
        return dummy.next
class Solution(object):
    def removeNthFromEnd(self, head, n):
        """
        :type head: ListNode
        :type n: int
        :rtype: ListNode
        """
        '''方法一：快慢指针'''
        p = ListNode(-1)
        p.next = head
        current = p
        last = head
        while n-1:
            last = last.next
            n = n-1
        while last.next:
            last = last.next
            current = current.next
        current.next = current.next.next
        return p.next
if __name__ == '__main__':
    head = [1, 2, 3, 4, 5]
    # head = [1]
    createlist = LinkList()
    list = createlist.linklist(head)
    n = 2
    remove = Solution()
    result = remove.removeNthFromEnd(list, n)
    print(result)

方法二：递归

这个方法是不断调用自身方法，寻找最后一个结点。当寻找到最后一个结点时开始反向计数，如果结点个数等于n，则返回当前结点的后续所有结点用于链接新的链表，如果不等于n，则返回所有结点。完整代码如下：

'''给你一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点。 '''
class ListNode(object):
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next
class LinkList(object):
    def __init__(self):
        self.head = None
    def linklist(self, data):
        if not data: return
        dummy = ListNode(-1)
        p = dummy
        p.next = ListNode(data[0])
        p = p.next
        for i in data[1:]:
            p.next =ListNode(i)
            p = p.next
        return dummy.next
class Solution(object):
    def removeNthFromEnd(self, head, n):
        """
        :type head: ListNode
        :type n: int
        :rtype: ListNode
        """
        '''方法二：递归'''
        if not head:
            self.count = 0
            return head
        head.next = self.removeNthFromEnd(head.next, n)
        self.count = self.count + 1
        return head.next if self.count == n else head
if __name__ == '__main__':
    head = [1, 2, 3, 4, 5]
    # head = [1]
    createlist = LinkList()
    list = createlist.linklist(head)
    n = 2
    remove = Solution()
    result = remove.removeNthFromEnd(list, n)
    print(result)

可以看出，使用递归的方法代码更加简洁。希望自己以后能够加强对递归方法的运用，使得自己的代码做到简洁，可读性强。