import { assertEquals } from "../deps.ts";
import Trie from "./index.ts";
import Trie2 from "./Trie.ts";

Deno.test("implement-trie-prefix-tree-1", () => {
    const trie = Trie();
    trie.insert("apple");
    assertEquals(true, trie.search("apple")); // 返回 True
    assertEquals(false, trie.search("app")); // 返回 False
    assertEquals(true, trie.startsWith("app")); // 返回 True
    trie.insert("app");
    assertEquals(true, trie.search("app"));
    assertEquals(true, trie.startsWith("ap")); // 返回 True
    assertEquals(true, trie.startsWith("app")); // 返回 True
    assertEquals(false, trie.startsWith("annnnpp")); // 返回 True
});
Deno.test("implement-trie-prefix-tree-2", () => {
    const trie = new Trie2();
    trie.insert("apple");
    assertEquals(true, trie.search("apple")); // 返回 True
    assertEquals(false, trie.search("app")); // 返回 False
    assertEquals(true, trie.startsWith("app")); // 返回 True
    trie.insert("app");
    assertEquals(true, trie.search("app"));
    assertEquals(true, trie.startsWith("ap")); // 返回 True
    assertEquals(true, trie.startsWith("app")); // 返回 True
    assertEquals(false, trie.startsWith("annnnpp")); // 返回 True
});