Trie樹提供給了一種能夠在字串的長度n時間內判斷出來是否在已有集合中已經存在這個字串了。

1056是判斷字首碼的問題。如果所有字串都不是其他的字串的字首的話，那麼就是可以直接編碼的。

java程式碼如下：

其中TrieTree類提供了一種建立字典樹的方法。在字典樹的建立過程中，可以同時求出是否已經存在該字串。

程式碼中註釋的比較清楚了。

import java.util.Scanner; public class Main { public static void main(String ars[]) { int count = 0; // File file = new File("D://test.txt"); Scanner cin = new Scanner(System.in); String bin = cin.nextLine(); while (true) { count++; TrieTree tree = new TrieTree(); boolean flag = true; boolean print = false; while (!bin.equals("9")) { if (flag) flag = tree.addWord(bin); if (!flag && !print) {// 如果是字首，輸出no System.out.println("Set " + count + " is not immediately decodable"); flag = false; print = true; } bin = cin.nextLine(); } if (flag) {// 如果不是字首，輸出yes System.out .println("Set " + count + " is immediately decodable"); } // 下一組測試 bin = cin.nextLine(); } } } class TrieTree { private Vertex root;// 一個Trie樹有一個根節點 // 內部類 protected class Vertex {// 節點類 protected Vertex[] edges;// 每個節點包含10個子節點(型別為自身) boolean isEnd;// 說明是不是一個字串的末尾 Vertex() { isEnd = false; edges = new Vertex[10]; for (int i = 0; i < edges.length; i++) { edges[i] = null; } } } public TrieTree() { root = new Vertex(); } public boolean addWord(String word) { if (word.length() == 1) return addWord(root, word, true); else return addWord(root, word, false); } // last標記word是不是新加入的字串的最後一位，true為是最後一位，false為不是最後一位 private boolean addWord(Vertex vertex, String word, boolean last) { int index = Integer.parseInt(word.charAt(0) + ""); if (last) { // 如果是新字串的最後一個元素，則設定為isEnd為true if (vertex.edges[index] == null) { // if the edge does NOT exist vertex.edges[index] = new Vertex(); } else { // 是新字串的最後一個數據，但是該元素在trie樹中已經存在，則肯定return false return false; } vertex.edges[index].isEnd = true;// 如果是最後一個元素，但是trie樹中還沒有該資料，設定為true return true; } else { if (vertex.edges[index] == null) { // if the edge does NOT exist vertex.edges[index] = new Vertex(); } if (vertex.edges[index].isEnd) // 如果找到了一個電話號碼是字首，輸出no return false; if (word.length() == 2) return addWord(vertex.edges[index], word.substring(1), true); // 下一個字元 else return addWord(vertex.edges[index], word.substring(1), false); // 下一個字元 } } }