fix for replacing single-char-node with value with array-node

wizzardo · Nov 13, 2023 · f7a4421 · f7a4421
1 parent dcc072c
commit f7a4421
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/modules/tools-misc/src/main/java/com/wizzardo/tools/misc/CharTree.java b/modules/tools-misc/src/main/java/com/wizzardo/tools/misc/CharTree.java
@@ -271,6 +271,7 @@ public CharTreeNode<V> append(char b) {
             if (next != null && this.b != b) {
                 ArrayCharTreeNode<V> node = new ArrayCharTreeNode<V>(Math.max(this.b, b));
                 node.set(this.b, next);
+                node.value = value;
                 node.append(b);
                 return node;
             } else if (this.b == b)
@@ -287,6 +288,7 @@ public CharTreeNode<V> set(char b, CharTreeNode<V> n) {
             if (next != null && this.b != b) {
                 ArrayCharTreeNode<V> node = new ArrayCharTreeNode<V>(Math.max(this.b, b));
                 node.set(this.b, next);
+                node.value = value;
                 node.set(b, n);
                 return node;
             } else if (this.b == b) {

diff --git a/src/test/java/com/wizzardo/tools/misc/CharTreeTest.java b/src/test/java/com/wizzardo/tools/misc/CharTreeTest.java
@@ -33,11 +33,12 @@ public void test2() {
                 .append("foo", "foo")
                 .append("bar", "bar")
                 .append("foobar", "foobar")
-                .append("foo", "foo");
+                .append("foo2bar", "foo2bar");
 
         Assert.assertEquals("foo", tree.get("foo".toCharArray()));
         Assert.assertEquals("bar", tree.get("bar".toCharArray()));
         Assert.assertEquals("foobar", tree.get("foobar".toCharArray()));
+        Assert.assertEquals("foo2bar", tree.get("foo2bar".toCharArray()));
     }
 
     @Test