Create Pigeonhole_Sort.py

Author: KavinduDr

Data Structures and Algorithms/Sorting Algorithms/Pigeonhole_Sort.py

@@ -0,0 +1,33 @@
+# Python program to implement Pigeonhole Sort
+
+def pigeonhole_sort(a):
+	# size of range of values in the list 
+	# (ie, number of pigeonholes we need)
+	my_min = min(a)
+	my_max = max(a)
+	size = my_max - my_min + 1
+
+	# our list of pigeonholes
+	holes = [0] * size
+
+	# Populate the pigeonholes.
+	for x in a:
+		assert type(x) is int, "integers only"
+		holes[x - my_min] += 1
+
+	# Put the elements back into the array in order.
+	i = 0
+	for count in range(size):
+		while holes[count] > 0:
+			holes[count] -= 1
+			a[i] = count + my_min
+			i += 1
+			
+
+a = [8, 1, 2, 7, 4, 5, 8]
+print("Sorted order is : ", end = ' ')
+
+pigeonhole_sort(a)
+		
+for i in range(0, len(a)):
+	print(a[i], end = ' ')