chore(Python): add count-sort (#587)

pull/632/head
Akanksha Shukla 2021-11-01 18:35:31 +05:30 committed by GitHub
parent 79efa7a0dc
commit a88e14cb8a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 29 additions and 0 deletions

View File

@ -32,6 +32,7 @@
## Sorting
- [Bubble Sort](sorting/bubble_sort.py)
- [Comb Sort](sorting/comb_sort.py)
- [Count Sort](sorting/count-sort.py)
- [Insertion Sort](sorting/insertion_sort.py)
- [Quicksort](sorting/quicksort.py)
- [Selection Sort](sorting/selection_sort.py)

View File

@ -0,0 +1,28 @@
#Code in python for count sort
def count_sort(a):
max_element = int(max(a))
min_element = int(min(a))
range_of_elements = max_element - min_element + 1
# Create a count array to store count of individual
# elements and initialize count array as 0
c= [0 for _ in range(range_of_elements)]
o= [0 for _ in range(len(a))]
for i in range(0, len(a)):
c[a[i]-min_element] += 1
for i in range(1, len(c)):
c[i] += c[i-1]
for i in range(len(a)-1, -1, -1):
o[c[a[i] - min_element] - 1] = a[i]
c[a[i] - min_element] -= 1
for i in range(0, len(a)):
a[i] = o[i]
return a
a=[-1,-2,6,5,10]
ans = count_sort(a)
print("Sorted array is ")
print(ans)
#TIME COMPLEXITY
#O(n+k) where n is the number of elements in input array and k is the range of input