Я пытаюсь реализовать очень простой Trie в Java, который поддерживает 3 операции. Мне бы хотелось, чтобы у него был метод вставки, метод has (то есть, это определенное слово в дереве) и метод toString для возврата дерева в виде строки. Я полагаю, что у меня вставка работает правильно, но hasString и toString оказываются трудными. Вот что у меня есть.
Три класса.
public class CaseInsensitiveTrie implements SimpleTrie {
//root node
private TrieNode r;
public CaseInsensitiveTrie() {
r = new TrieNode();
}
public boolean has(String word) throws InvalidArgumentUosException {
return r.has(word);
}
public void insert(String word) throws InvalidArgumentUosException {
r.insert(word);
}
public String toString() {
return r.toString();
}
public static void main(String[] args) {
CaseInsensitiveTrie t = new CaseInsensitiveTrie();
System.out.println("Testing some strings");
t.insert("TEST");
t.insert("TATTER");
System.out.println(t.has("TEST"));
}
}
И узел класса
public class TrieNode {
//make child nodes
private TrieNode[] c;
//flag for end of word
private boolean flag = false;
public TrieNode() {
c = new TrieNode[26]; //1 for each letter in alphabet
}
protected void insert(String word) {
int val = word.charAt(0) - 64;
//if the value of the child node at val is null, make a new node
//there to represent the letter
if (c[val] == null) {
c[val] = new TrieNode();
}
//if word length > 1, then word is not finished being added.
//otherwise, set the flag to true so we know a word ends there.
if (word.length() > 1) {
c[val].insert(word.substring(1));
} else {
c[val].flag = true;
}
}
public boolean has(String word) {
int val = word.charAt(0) - 64;
if (c[val]!=null && word.length()>1) {
c[val].has(word.substring(1));
} else if (c[val].flag==true && word.length()==1) {
return true;
}
return false;
}
public String toString() {
return "";
}
}
Таким образом, в основном, при создании Trie, TrieNode создается как корень с 26 дочерними элементами. При попытке вставки вставка вызывается на этом корневом узле, который рекурсивно создает новый узел в правильной позиции и продолжается до тех пор, пока слово не будет завершено. Я считаю, что метод работает правильно.
Моя функция has очень нарушена, потому что у меня есть , чтобы по какой-то причине этот оператор возврата был вне скобок. Я не могу содержать его в предложении else, иначе компилятор жалуется. Кроме этого, я думаю, что этот метод должен работать с некоторыми изменениями, но я не могу понять это для моей жизни.
toString - зверь, которого я пытался решить, но ничто из того, что я выбрасываю, не работает, так что я оставлю это так, пока не решу проблему. Если я получу работу, я могу найти способ переформатировать ее в функцию toString.
Цель int val = word.charAt (0) - 64; это потому, что каждая введенная строка должна иметь все заглавные буквы (я создам функцию форматирования строки, чтобы гарантировать это позже), поэтому значение int первой буквы - 64 будет его позицией в массиве. то есть индекс массива 0 - это A, поэтому A = 64, A - 64 = 0. B = 65, B - 64 = 1 и т. д.