Given a linked list, remove the n-th node from the end of list and return its head.
Example:
Given linked list: 1->2->3->4->5, and n = 2. After removing the second node from the end, the linked list becomes 1->2->3->5.
Note:
Given n will always be valid.
Follow up:
Could you do this in one pass?
My solution:
# 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
"""
fast = slow = head
for _ in range(n):
fast = fast.next
if not fast:
return head.next
while fast.next:
fast = fast.next
slow = slow.next
print(fast.val, slow.val)
slow.next = slow.next.next
return head
'DataStructure > Linked List' 카테고리의 다른 글
Reverse Linked List (0) | 2018.10.26 |
---|---|
Delete Node in a Linked List (0) | 2018.10.25 |
Print the Elements of a Linked List (0) | 2018.05.08 |