본문 바로가기

DataStructure/Linked List

Remove Nth Node From End of List

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