Change: print only the 10 most common words

1 files changed, 11 insertions, 2 deletions

diff --git a/src/main.rs b/src/main.rs
index 4045cd7..2162453 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -12,7 +12,7 @@ fn main() -> io::Result<()> {
         let mut f = File::open(&filename)?;
         let mut reader = io::BufReader::new(f);
         let counts = count_words(&mut reader)?;
-        counts.print()
+        counts.print(10)
     }
     Ok(())
 }
@@ -69,8 +69,17 @@ impl WordCounts {
         *counter += count;
     }
 
-    fn print(&self) {
+    fn print(&self, max: usize) {
+        let mut top = Vec::new();
         for (word, count) in self.counts.iter() {
+            top.push((count, word));
+            if top.len() > max {
+                top.sort();
+                top.reverse();
+                top.truncate(max);
+            }
+        }
+        for (count, word) in top.iter() {
             println!("{} {}", count, word);
         }
     }