Added the counting sort method which uses stream API

Changed the original sorting method: creating and throwing an exception it costs then using getOrDefault for a map

Sorts/CountingSort.java

@@ -1,10 +1,17 @@
 import java.util.ArrayList;
+import java.util.List;
 import java.util.Map;
 import java.util.TreeMap;
+import java.util.stream.IntStream;
+import java.util.stream.Stream;
+
+import static java.util.stream.Collectors.toList;
+import static java.util.stream.Collectors.toMap;
 
 /**
  *
  * @author Youssef Ali (https://github.com/youssefAli11997)
+ * @author Podshivalov Nikita (https://github.com/nikitap492)
  *
  */
 
@@ -13,26 +20,19 @@ class CountingSort {
     /**
      * This method implements the Generic Counting Sort
      *
-     * @param array The array to be sorted 
+     * @param list The list to be sorted
-     * @param last The count of total number of elements in array
+     *
-     * Sorts the array in increasing order
+     * Sorts the list in increasing order
-     * It uses array elements as keys in the frequency map
+     * The method uses list elements as keys in the frequency map
      **/
+    public static <T extends Comparable<T>> List<T> CS(List<T> list) {
 
-    public static <T extends Comparable<T>> void CS(T[] array, int last) {
+        Map<T, Integer> frequency = new TreeMap<>();
-    
-    	Map<T, Integer> frequency = new TreeMap<T, Integer>();
         // The final output array
-    	ArrayList<T> sortedArray = new ArrayList<T>();
+        List<T> sortedArray = new ArrayList<>(list.size());
 
         // Counting the frequency of @param array elements
-    	for(T t : array) {
+        list.forEach(v -> frequency.put(v, frequency.getOrDefault(v, 0) + 1));
-    		try{
-    			frequency.put(t, frequency.get(t)+1);
-    		}catch(Exception e){ // new entry
-    			frequency.put(t, 1);
-    		}
-        }
 
         // Filling the sortedArray
         for(Map.Entry<T, Integer> element : frequency.entrySet()) {
@@ -40,51 +40,67 @@ class CountingSort {
                 sortedArray.add(element.getKey());
         }
 
-    	for(int i=0; i<array.length; i++){
+        return sortedArray;
-    		array[i] = sortedArray.get(i);
     }
+
+
+    /**
+     * Stream Counting Sort
+     * The same as method {@link CountingSort#CS(List)} but this method uses stream API
+     *
+     * @param list The list to be sorted
+     *
+     **/
+    public static <T extends Comparable<T>> List<T> streamCS(List<T> list) {
+        return  list.stream()
+                .collect(toMap(k -> k, v -> 1, (v1, v2) -> v1 + v2, TreeMap::new))
+                .entrySet()
+                .stream()
+                .flatMap(entry -> IntStream.rangeClosed(1, entry.getValue()).mapToObj(t -> entry.getKey()))
+                .collect(toList());
     }
 
     // Driver Program
     public static void main(String[] args) {
         // Integer Input
-        Integer[] arr1 = {4,23,6,78,1,54,231,9,12};
+        List<Integer> unsortedInts = Stream.of(4, 23, 6, 78, 1, 54, 23, 1, 9, 231, 9, 12).collect(toList());
-        int last = arr1.length;
 
         System.out.println("Before Sorting:");
-        for (int i=0;i<arr1.length;i++) {
+        printList(unsortedInts);
-            System.out.print(arr1[i] + " ");
-        }
-        System.out.println();
 
-        CS(arr1, last);
+        // Output => 1 1 4 6 9 9 12 23 23 54 78 231
-
-        // Output => 1 4 6 9 12 23 54 78 231
         System.out.println("After Sorting:");
-        for (int i=0;i<arr1.length;i++) {
+        printList(CS(unsortedInts));
-            System.out.print(arr1[i] + " ");
+        System.out.println("After Sorting By Streams:");
-        }
+        printList(streamCS(unsortedInts));
-        System.out.println();
 
-        System.out.println("------------------------------");
+        System.out.println("\n------------------------------\n");
 
         // String Input
-        String[] array1 = {"c", "a", "e", "b","d"};
+        List<String> unsortedStrings = Stream.of("c", "a", "e", "b","d", "a", "f", "g", "c").collect(toList());
-        last = array1.length;
 
         System.out.println("Before Sorting:");
-        for (int i=0;i<array1.length;i++) {
+        printList(unsortedStrings);
-            System.out.print(array1[i] + " ");
-        }
-        System.out.println();
 
-        CS(array1, last);
+        //Output => a a b c c d e f g
-
-        //Output => a	  b	 c	d	e
         System.out.println("After Sorting:");
-        for(int i=0; i<last; i++) {
+        printList(CS(unsortedStrings));
-            System.out.print(array1[i]+" ");
+
+        System.out.println("After Sorting By Streams:");
+        printList(streamCS(unsortedStrings));
+
     }
 
+    /**
+     * Just print list
+     * @param toPrint - a list which should be printed
+     */
+    private static void printList(List<?> toPrint){
+        toPrint.stream()
+                .map(Object::toString)
+                .map(str -> str + " ")
+                .forEach(System.out::print);
+
+        System.out.println();
     }
 }