fixed trie

userr2232 · Oct 14, 2021 · 5836588 · 5836588
1 parent f65aa68
commit 5836588
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Study/trie.cc b/Study/trie.cc
@@ -13,9 +13,9 @@ void add_string(string s) {
     for(int i = 0; i < n; ++i) {
         string_counter[node]++;
         int ord = s[i] - 'a';
-        int next_node = trie[node][ord];
-        if(next_node == 0) trie[node][ord] = ++nodeCounter, ends_here[next_node = nodeCounter] = true;
-        else if(i == n-1) ends_here[node] = true;
+        int& next_node = trie[node][ord];
+        if(next_node == 0) next_node = ++nodeCounter;
+        if(i == n-1) ends_here[node] = true;
         node = next_node;
     }
     string_counter[node]++;