Design Add and Search Words Data Structure - addWord() 코드 리팩토링

forest000014 · forest000014 · commit f848beb03dbc · 2025-01-15T00:32:08.000+09:00
diff --git a/design-add-and-search-words-data-structure/forest000014.java b/design-add-and-search-words-data-structure/forest000014.java
@@ -38,12 +38,10 @@ public void addWord(String word) {
                 curr.children.put(ch, new Node(ch));
             }
 
-            if (i == word.length() - 1) {
-                curr.children.get(ch).ends = true;
-            } else {
-                curr = curr.children.get(ch);
-            }
+            curr = curr.children.get(ch);
         }
+
+        curr.ends = true;
     }
 
     public boolean search(String word) {

Original file line number	Diff line number	Diff line change
`@@ -38,12 +38,10 @@ public void addWord(String word) {`
`38`	`38`	`curr.children.put(ch, new Node(ch));`
`39`	`39`	`}`
`40`	`40`
`41`		`- if (i == word.length() - 1) {`
`42`		`- curr.children.get(ch).ends = true;`
`43`		`- } else {`
`44`		`- curr = curr.children.get(ch);`
`45`		`- }`
	`41`	`+ curr = curr.children.get(ch);`
`46`	`42`	`}`
	`43`	`+`
	`44`	`+ curr.ends = true;`
`47`	`45`	`}`
`48`	`46`
`49`	`47`	`public boolean search(String word) {`