exo5: add a SortedMap from the TreeMap.

[Project_POO.git] / exo5 / Main.java

diff --git a/exo5/Main.java b/exo5/Main.java
index 4075cbc4ed5997f2f9c18bf6b6ec1b59a5fba719..1ec751215c7c5bd4710ca47cefe6a69f39c6a297 100644 (file)
--- a/exo5/Main.java
+++ b/exo5/Main.java
@@ -1,3 +1,8 @@
+import java.io.File;
+import java.util.Scanner;
+import java.io.IOException;
+import java.util.TreeMap;
+import java.util.SortedMap;
 
 class Main {
 
@@ -6,6 +11,36 @@ class Main {
      * @param String[] args main() function arguments array
      */
     public static void main(String[] args) {
+        //TreeMap<String, Integer> tm = new TreeMap<String, Integer>(new StrComparator());
+        // default TreeMap comparator keep alphabetical order
+        TreeMap<String, Integer> tm = new TreeMap<String, Integer>();
+        File f = new File("test_file.txt");
+        try {
+            Scanner sc = new Scanner(f);
+            while (sc.hasNext()) {
+                String wordCursor = sc.next();
+                //NOTE: words comparaison is case sensitive and punctuation aware
+                Integer count = tm.get(wordCursor);
+                if (count == null) {
+                    count = 0;
+                }
+                tm.put(wordCursor, count + 1);
+            }
+            sc.close();
+        }
+        catch (IOException e) {
+            e.printStackTrace();
+        }
 
+        System.out.println("TreeMap full content:");
+        for (String wordCursor : tm.keySet()) {
+            System.out.println("Word \"" + wordCursor + "\" occured " + tm.get(wordCursor) + " times");
+        }
+
+        System.out.println("SortedMap content after the word \"milieu\":");
+        SortedMap<String, Integer> stm = tm.tailMap("milieu");
+        for (String wordCursor : stm.keySet()) {
+            System.out.println("Word \"" + wordCursor + "\" occured " + stm.get(wordCursor) + " times");
+        }
     }
 }