Python Program - Counting Sort

Counting sort is based on the idea that the number of occurrences of distinct elements is counted and stored in another array (say frequency array), by mapping the value of the distinct elements with index numbers of the array. The frequency array is then iterated over to use distinct elements and their occurrences and put them into the output array in a sorted way.

Example:

To understand the counting sort, lets consider an unsorted array A = [9, 1, 2, 5, 9, 9, 2, 1, 3, 3] and discuss each step taken to sort the array in ascending order.

Step 1: In this step, the largest element of the A[ ] is found out which is (9) and an another array called frequency is initiated with size [max(A[ ])+1]. Then the array A[ ] is iterated over to store the number of occurrences of each distinct element in frequency array, by mapping the distinct elements with index number of frequency array.

Step 2: In this step, the frequency array is iterated over to get the information about each distinct element and its occurrences which is further used to build the sorted array.

Counting sort is used most efficiently when the range of input elements is not significantly larger than the number of elements to be sorted. Along with this, the same concept can be used with negative input data as well.

Implementation of Counting Sort

# function for counting sort
def countingsort(MyList):
  n = len(MyList)
  max = 0
  #find largest element in the Array
  for i in MyList:
    if max < i:
      max = i

  #Create a freq array to store number of occurrences of 
  #each unique elements in the given list
  freq = [0 for i in range(0,max+1)] 
  for i in range(0,n):
    freq[MyList[i]] += 1

  #sort the given array using freq list
  j=0
  for i in range(0, max+1): 
    while freq[i]>0:
      MyList[j] = i
      j +=1
      freq[i] -= 1 

# function to print list
def PrintList(MyList):
  for i in MyList:
    print(i, end=" ")
  print("\n")
  
# test the code                 
MyList = [9, 1, 2, 5, 9, 9, 2, 1, 3, 3]
print("Original List")
PrintList(MyList)

countingsort(MyList)
print("Sorted List")
PrintList(MyList)

The above code will give the following output:

Original List
9 1 2 5 9 9 2 1 3 3 

Sorted List
1 1 2 2 3 3 5 9 9 9 

Time Complexity:

The time complexity to iterate over the input data is Θ(N), where N is the number of elements in unsorted array and the time complexity to iterate over the frequency array is Θ(K), where K is the range of input data. The overall time complexity of counting sort is Θ(N+K) in all cases.