-
Notifications
You must be signed in to change notification settings - Fork 0
/
141_linked_list_cycle.py
82 lines (63 loc) · 1.9 KB
/
141_linked_list_cycle.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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
'''
141. Linked List Cycle
Given a linked list, determine if it has a cycle in it.
To represent a cycle in the given linked list, we use an integer pos which represents the position (0-indexed) in the linked list where tail connects to. If pos is -1, then there is no cycle in the linked list.
Example 1:
Input: head = [3,2,0,-4], pos = 1
Output: true
Explanation: There is a cycle in the linked list, where tail connects to the second node.
Example 2:
Input: head = [1,2], pos = 0
Output: true
Explanation: There is a cycle in the linked list, where tail connects to the first node.
Example 3:
Input: head = [1], pos = -1
Output: false
Explanation: There is no cycle in the linked list.
Follow up:
Can you solve it using O(1) (i.e. constant) memory?
'''
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def hasCycle(self, head):
"""
approach:
1. dictionary
2. 2 pointers fast and slow
"""
# 1 dictionary
seen = {}
# can also use set
while head:
if head in seen:
return True
else:
seen[head] = 0
head = head.next
return False
# 2
fast = slow = head
while fast and fast.next:
fast = fast.next.next
slow = slow.next
if slow is fast:
return True
return False
# alternative Easier to ask for forgiveness than permission coding style ~slightly faster
try:
slow = head
fast = head.next
while slow is not fast:
slow = slow.next
fast = fast.next.next
return True
except:
return False
if __name__ == '__main__':
# begin
s = Solution()
print(s.hasCycle(1, 2))