Merge remote-tracking branch 'upstream/master'

data-structures/src/main/java/com/baeldung/trie/Trie.java

@@ -1,13 +1,13 @@
 package com.baeldung.trie;
 
-public class Trie {
+class Trie {
     private TrieNode root;
 
     Trie() {
         root = new TrieNode();
     }
 
-    public void insert(String word) {
+    void insert(String word) {
         TrieNode current = root;
 
         for (int i = 0; i < word.length(); i++) {
@@ -16,11 +16,11 @@ public class Trie {
         current.setEndOfWord(true);
     }
 
-    public boolean delete(String word) {
+    boolean delete(String word) {
         return delete(root, word, 0);
     }
 
-    public boolean containsNode(String word) {
+    boolean containsNode(String word) {
         TrieNode current = root;
 
         for (int i = 0; i < word.length(); i++) {
@@ -34,7 +34,7 @@ public class Trie {
         return current.isEndOfWord();
     }
 
-    public boolean isEmpty() {
+    boolean isEmpty() {
         return root == null;
     }
 
@@ -51,7 +51,7 @@ public class Trie {
         if (node == null) {
             return false;
         }
-        boolean shouldDeleteCurrentNode = delete(node, word, index + 1);
+        boolean shouldDeleteCurrentNode = delete(node, word, index + 1) && !node.isEndOfWord();
 
         if (shouldDeleteCurrentNode) {
             current.getChildren().remove(ch);

data-structures/src/main/java/com/baeldung/trie/TrieNode.java

@@ -4,28 +4,18 @@ import java.util.HashMap;
 import java.util.Map;
 
 class TrieNode {
-    private Map<Character, TrieNode> children;
+    private final Map<Character, TrieNode> children = new HashMap<>();
     private boolean endOfWord;
 
-    public TrieNode() {
-        children = new HashMap<>();
-        endOfWord = false;
-    }
-
-    public Map<Character, TrieNode> getChildren() {
+    Map<Character, TrieNode> getChildren() {
         return children;
     }
 
-    public void setChildren(Map<Character, TrieNode> children) {
-        this.children = children;
-    }
-
-    public boolean isEndOfWord() {
+    boolean isEndOfWord() {
         return endOfWord;
     }
 
-    public void setEndOfWord(boolean endOfWord) {
+    void setEndOfWord(boolean endOfWord) {
         this.endOfWord = endOfWord;
     }
-
 }