-
Notifications
You must be signed in to change notification settings - Fork 62
/
delete_nth_node.py
37 lines (30 loc) · 977 Bytes
/
delete_nth_node.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# 删除链表倒数第N个节点
class ListNode:
def __init__(self, val=0, next=None):
self.val = val
self.next = next
class Solution:
# 先计数,再删除
def removeNthFromEnd_1(self, head: ListNode, n: int) -> ListNode:
def length(node: ListNode) -> int:
k = 0
while node:
k, node = k + 1, node.next
return k
dummy = ListNode(0, head)
length = length(head)
cur = dummy
for _ in range(1, length - n + 1):
cur = cur.next
cur.next = cur.next.next
return dummy.next
# 快慢指针,一次遍历
def removeNthFromEnd_2(self, head: ListNode, n: int) -> ListNode:
dummy = ListNode(0, head)
slow, fast = dummy, head
for _ in range(n):
fast = fast.next
while fast:
slow, fast = slow.next, fast.next
slow.next = slow.next.next
return dummy.next