1. 程式人生 > >leetcode19 刪除連結串列的倒數第N個節點

leetcode19 刪除連結串列的倒數第N個節點

給定一個連結串列,刪除連結串列的倒數第 n 個節點,並且返回連結串列的頭結點。

示例:

給定一個連結串列: 1->2->3->4->5, 和 n = 2.

當刪除了倒數第二個節點後,連結串列變為 1->2->3->5.
說明:

給定的 n 保證是有效的。

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

class Solution:
    def removeNthFromEnd(self, head, n):
        """
        :type head: ListNode
        :type n: int
        :rtype: ListNode
        """
        if head.next == None:
            return []
        else:
            pre = head
            end = head
            for i in range(n):
                end = end.next
            if not end:
                return head.next
            
            while end.next:
                pre = pre.next
                end = end.next
            pre.next = pre.next.next
            return head

在這裡插入圖片描述