repair a bug

baidu · Dec 30, 2020 · 29f0fd2 · 29f0fd2
1 parent 85ee86a
commit 29f0fd2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/LAC/prefix_tree.py b/python/LAC/prefix_tree.py
@@ -49,7 +49,7 @@ def search(self, content):
         result = []
         length = len(content)
         for start in range(length):
-            for end in range(start+1, length):
+            for end in range(start+1, length+1):
                 pos = self.tree.get(content[start:end], -1)
                 if pos == -1:
                     break
@@ -71,7 +71,7 @@ def search_all(self, content):
 
         length = len(content)
         for start in range(length):
-            for end in range(start+1, length):
+            for end in range(start+1, length+1):
                 pos = self.tree.get(content[start:end], -1)
                 if pos == -1:
                     break