forked from hating/IntroductionToAlgorithms
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathMergeSort.py
More file actions
42 lines (36 loc) · 976 Bytes
/
MergeSort.py
File metadata and controls
42 lines (36 loc) · 976 Bytes
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
import random
def merge_sort(A):
def merge_sort_recursive(A, p, r):
if p < r:
q = (p + r) // 2
merge_sort_recursive(A, p, q)
merge_sort_recursive(A, q + 1, r)
merge(A, p, q, r)
def merge(A, p, q, r):
L = A[p:q+1]
R = A[q+1:r+1]
i = j = 0
k = p
while i < len(L) and j < len(R):
if L[i] <= R[j]:
A[k] = L[i]
i += 1
else:
A[k] = R[j]
j += 1
k += 1
while i < len(L):
A[k] = L[i]
i += 1
k += 1
while j < len(R):
A[k] = R[j]
j += 1
k += 1
merge_sort_recursive(A, 0, len(A) - 1)
if __name__ == "__main__":
print("Now displaying MergeSort")
A = [random.randint(0, 1000) for _ in range(random.randint(5, 100))]
print("Before:", A)
merge_sort(A)
print("After:", A)