From 3f3bebdc9561a78950f3979748a3abc3b2315c32 Mon Sep 17 00:00:00 2001
From: iovoi <iovoi@users.noreplay.github.com>
Date: Sun, 10 Oct 2021 23:05:01 +0800
Subject: [PATCH] correct the bug that removes a prefix word when removing a
 longer word

---
 src/com/interview/suffixprefix/Trie.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/com/interview/suffixprefix/Trie.java b/src/com/interview/suffixprefix/Trie.java
index d91dc8f1..5077161c 100644
--- a/src/com/interview/suffixprefix/Trie.java
+++ b/src/com/interview/suffixprefix/Trie.java
@@ -140,7 +140,7 @@ private boolean delete(TrieNode current, String word, int index) {
         if (shouldDeleteCurrentNode) {
             current.children.remove(ch);
             //return true if no mappings are left in the map.
-            return current.children.size() == 0;
+            return (current.children.size() == 0 && !current.endOfWord);
         }
         return false;
     }