Simplify multiset add_elem method

vitorenesduarte · Mar 9, 2019 · e2b5577 · e2b5577
1 parent 13c725b
commit e2b5577
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 11 deletions.
diff --git a/Makefile b/Makefile
@@ -1,5 +1,5 @@
 fmt:
-	rustfmt tests/* benches/* src/*
+	rustfmt src/* tests/* benches/*
 
 coverage:
 	cargo coverage

diff --git a/src/multiset.rs b/src/multiset.rs
@@ -89,16 +89,9 @@ impl<T: Ord> MultiSet<T> {
     /// assert_eq!(mset.count(&17), 1);
     /// ```
     pub fn add_elem(&mut self, elem: T) {
-        match self.occurrences.get_mut(&elem) {
-            Some(count) => {
-                // if the element exists, increase its count
-                *count += 1;
-            }
-            None => {
-                // otherwise, insert it
-                self.occurrences.insert(elem, 1);
-            }
-        }
+        // increase element count
+        let count = self.occurrences.entry(elem).or_insert(0);
+        *count += 1;
     }
 
     /// Returns the number of occurrences of an element.