题目:
Given a linked list, remove the
n
-th node from the end of list and return its head.
代码:
# 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
"""
i=1
node = head
while node.next:
i+=1
node=node.next
if i-n==0:
if head.next:
nxt=head.next
del(head)
head = nxt
else:
head=None
return head
j=1
node=head
while j<i-n:
node=node.next
j+=1
nxt=node.next
node.next=nxt.next
del(nxt)
return head
解释:该题是根据给出的数字n删除链表倒数第n个元素。
解题时首先计算出了链表共有i个元素,倒数第n个元素的前一个元素为第i-n个元素node,如果i-n为0,则删除的是第一个元素,需要将头指针指向下一各元素,同时,如果只有一个元素则头元素为空。否则删除的元素为第i-n+1个元素nxt,node的下一个指针指向nxt所指向的。