Last active
June 5, 2019 13:03
-
-
Save Kenny2github/9a0c103ce5631cad947b8d14c4f6e28b to your computer and use it in GitHub Desktop.
Vortex Sort - swirl elements into sorted order!
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
def vortex_sort(arr): | |
"""Swirl the array into order. | |
Guaranteed to only use < for comparison, so use any objects you like. | |
""" | |
arlen = len(arr) | |
rs = [] | |
mult = 1 | |
last = 0 | |
while last < arlen: | |
number = mult * 4 | |
mult += 2 | |
rs.append(arr[last:last + number]) | |
last += number | |
if len(rs) <= 1: | |
#bubble sort | |
arr = rs[0] | |
arlen = len(arr) | |
sortedq = False | |
srtd = 1 | |
while not sortedq: | |
sortedq = True | |
for i in range(arlen - srtd): | |
if arr[i+1] < arr[i]: | |
arr[i], arr[i+1] = arr[i+1], arr[i] | |
sortedq = False | |
srtd += 1 | |
return arr | |
sortedq = False | |
while not sortedq: | |
sortedq = True | |
for i in reversed(range(len(rs) - 1)): | |
for _ in range(len(rs[i])): | |
#swirl this ring to outer ring | |
idx = 1 + -8 // len(rs[i]) | |
olen = len(rs[i + 1]) | |
idx, min_val = min( | |
(idx, rs[i + 1][idx % olen]), | |
(idx + 1, rs[i + 1][(idx + 1) % olen]), | |
(idx + 2, rs[i + 1][(idx + 2) % olen]), | |
key=lambda k: k[1] | |
) | |
if rs[i+1][idx] < rs[i][0]: | |
rs[i][0], rs[i+1][idx] = rs[i+1][idx], rs[i][0] | |
sortedq = False | |
for r in rs: | |
r.insert(0, r.pop(-1)) | |
for i in range(len(rs)): | |
#recursively vortext sort ring | |
rs[i] = vortex_sort(rs[i]) | |
#merge rings together | |
rlen = len(rs) | |
while rlen > 1: | |
for i in range(0, rlen, 2): | |
j = i + 1 | |
if j >= rlen: | |
continue | |
_i = i | |
i, j = rs[i], rs[j] | |
len1 = len(i) | |
len2 = len(j) | |
tarlen = len(i) + len(j) | |
tmparr = [0] * tarlen | |
m, n, p = 0, 0, 0 | |
while m < len1 and n < len2 and p < tarlen: | |
if not j[n] < i[m]: | |
tmparr[p] = i[m] | |
m += 1 | |
else: | |
tmparr[p] = j[n] | |
n += 1 | |
p += 1 | |
if m < len1: | |
tmparr = tmparr[:p] | |
tmparr.extend(i[m:]) | |
elif n < len2: | |
tmparr = tmparr[:p] | |
tmparr.extend(j[n:]) | |
rs[_i] = tmparr | |
rs = [rs[i] for i in range(0, rlen, 2)] | |
rlen = len(rs) | |
return rs[0] | |
if __name__ == '__main__': | |
import random | |
a = list(range(int(input('How many elements? ')))) | |
random.shuffle(a) | |
print('start:', a) | |
b = vortex_sort(a) | |
print('\nend:', b) | |
print('\nsorted?', list(sorted(a)) == b) |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
def vortex_sort(arr, start=0, end=None): | |
"""Swirl the array into order. | |
Guaranteed to only use < for comparison, so use any objects you like. | |
""" | |
if not arr[start:end]: | |
return arr | |
arlen = (end or len(arr)) - start | |
rs = [] | |
mult = 1 | |
last = 0 | |
while last < arlen: | |
number = mult * 4 | |
mult += 2 | |
rs.append(( | |
last, #start | |
min(last + number, arlen) - last #length | |
)) #start, length | |
last += number | |
rlen = len(rs) | |
if rlen <= 1: | |
#bubble sort | |
sortedq = False | |
srtd = 1 | |
while not sortedq: | |
sortedq = True | |
for i in range(arlen - srtd): | |
if arr[i+1+start] < arr[i+start]: | |
arr[i+start], arr[i+1+start] = arr[i+1+start], arr[i+start] | |
sortedq = False | |
srtd += 1 | |
return arr | |
sortedq = False | |
while not sortedq: | |
sortedq = True | |
for i in reversed(range(rlen - 1)): | |
for _ in range(rs[i][1]): | |
#swirl this ring to outer ring | |
idx = 1 + -8 // rs[i][1] #2 = length | |
olen = rs[i + 1][1] | |
idx, min_val = min( | |
(idx % olen, arr[rs[i + 1][0] + idx % olen + start]), | |
((idx + 1) % olen, | |
arr[rs[i + 1][0] + (idx + 1) % olen + start]), | |
((idx + 2) % olen, | |
arr[rs[i + 1][0] + (idx + 2) % olen + start]), | |
key=lambda k: k[1] | |
) | |
if arr[rs[i+1][0] + idx + start] < arr[rs[i][0] + start]: | |
arr[rs[i][0] + start], arr[rs[i+1][0] + idx + start] \ | |
= arr[rs[i+1][0] + idx + start], \ | |
arr[rs[i][0] + start] | |
sortedq = False | |
for r in rs: | |
arr.insert(r[0] + start, arr.pop(r[0] + r[1] - 1 + start)) | |
for r in rs: | |
#recursively vortext sort ring | |
vortex_sort(arr, r[0] + start, r[0] + r[1] + start) | |
#merge rings together | |
while rlen > 1: | |
for i in range(0, rlen, 2): | |
if i + 1 >= rlen: | |
continue | |
len1 = rs[i][1] | |
len2 = rs[i + 1][1] | |
tarlen = len1 + len2 | |
tmparr = [] | |
m, n = 0, 0 | |
while m < len1 and n < len2: | |
if not arr[rs[i+1][0] + n + start] \ | |
< arr[rs[i][0] + m + start]: | |
tmparr.append(arr[rs[i][0] + m + start]) | |
m += 1 | |
else: | |
tmparr.append(arr[rs[i+1][0] + n + start]) | |
n += 1 | |
if m < len1: | |
tmparr.extend(arr[rs[i][0] + m + start | |
:rs[i][0] + rs[i][1] + start]) | |
elif n < len2: | |
tmparr.extend(arr[rs[i+1][0] + n + start | |
:rs[i+1][0] + rs[i+1][1] + start]) | |
arr[rs[i][0] + start:rs[i][0] + tarlen + start] = tmparr | |
rs = [(rs[i][0], rs[i][1] + rs[i+1][1]) | |
if i + 1 < rlen | |
else rs[i] | |
for i in range(0, rlen, 2)] | |
rlen = len(rs) | |
return arr | |
if __name__ == '__main__': | |
import random | |
a = list(range(int(input('How many elements? ')))) | |
random.shuffle(a) | |
print('start:', a) | |
b = vortex_sort(a) | |
print('\nend:', b) | |
print('\nsorted?', list(sorted(a)) == b) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment