From 5836588e482d24918e4b4ca6660a8154c5d9b66e Mon Sep 17 00:00:00 2001
From: userr2232 <reynaldo.rz.26@gmail.com>
Date: Wed, 13 Oct 2021 21:37:27 -0500
Subject: [PATCH] fixed trie

---
 Study/trie.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/Study/trie.cc b/Study/trie.cc
index e5b7c23..05d38f2 100644
--- 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]++;