- Notifications
You must be signed in to change notification settings - Fork 625
/
Copy path86.py
48 lines (40 loc) · 1.18 KB
/
86.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
'''
Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.
You should preserve the original relative order of the nodes in each of the two partitions.
Example:
Input: head = 1->4->3->2->5->2, x = 3
Output: 1->2->2->4->3->5
'''
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, x):
# self.val = x
# self.next = None
classSolution(object):
defpartition(self, head, x):
"""
:type head: ListNode
:type x: int
:rtype: ListNode
"""
ifnotheadornothead.next:
returnhead
left, right=ListNode(0), ListNode(0)
leftPtr, rightPtr=left, right
whilehead:
ifhead.val<x:
leftPtr.next=ListNode(head.val)
leftPtr=leftPtr.next
else:
rightPtr.next=ListNode(head.val)
rightPtr=rightPtr.next
head=head.next
ifnotleft.next:
returnright.next
elifnotright.next:
returnleft.next
else:
leftPtr.next=right.next
returnleft.next
# Time: O(N)
# Space: O(N)