题目描述
这个题目属于链表的常规方法,最常能想到的就是快慢指针法,由于本人对于递归的方法运用不熟练,因此下面介绍两种方法来解决这个问题。
方法一:快慢指针法
定义两个指针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)
可以看出,使用递归的方法代码更加简洁。希望自己以后能够加强对递归方法的运用,使得自己的代码做到简洁,可读性强。