trie 是一种像树数据结构一样存储字符串的数据结构。节点中的最大子节点数等于字母表的大小。人们可以轻松地按字母顺序打印字母,这是散列无法实现的。
Trie 的属性:
- 这是一棵多路树。
- 每个节点有1到N个子节点。
- 每个叶节点对应于存储的字符串,它是从根到其边的路径上的字符链。
特里的类型:
- 标准特里
- 后缀树
- 压缩树
压缩树:
尝试使用度数至少为 2 的节点。它是通过压缩标准树的节点来完成的。它也被称为基数尝试。用于实现空间优化
由于节点被压缩。让我们直观地比较标准树和压缩树的结构以获得更好的方法。在内存方面,压缩的 trie 树使用很少的节点数量,这提供了巨大的内存优势(特别是对于长的)具有长公共前缀的字符串。在速度方面,常规的特里树会稍微快一些,因为它的操作不涉及任何字符串操作,它们是简单的循环。
在下图中,左树是标准树,右树是压缩树。
执行:
标准的特里树节点如下所示:
Java
class node {
node[] children = new node[26];
boolean isWordEnd;
}
Java
class node {
node[] children = new node[26];
StringBuilder[] edgeLabel = new StringBuilder[26];
boolean isEnd;
}
Java
class CompressedNode {
int bitNumber;
int data;
CompressedNode leftChild, rightChild;
}
Java
class CompressedNode {
// Root Node
private CompressedNode root;
private static final int MaxBits = 10;
// Constructor
public CompressedNode() { root = null; }
// Function to check if empty
public boolean isEmpty() { return root == null; }
// Function to clear
public void makeEmpty() { root = null; }
}
Java
// Function to search a key k
// in the trie
public boolean search(int k)
{
// Find the number of bits
int numOfBits = (int)(Math.log(k) / Math.log(2)) + 1;
// If error occurs
if (numOfBits > MaxBits) {
System.out.println("Error : Number too large");
return false;
}
// Search Node
CompressedNode searchNode = search(root, k);
// If the data matches
if (searchNode.data == k)
return true;
// Else return false
else
return false;
}
Java
// Function to implement the insert
// functionality in the trie
private CompressedNode insert(
CompressedNode t, int ele)
{
CompressedNode current, parent;
CompressedNode lastNode, newNode;
int i;
// If Node is NULL
if (t == null) {
t = new CompressedNode();
t.bitNumber = 0;
t.data = ele;
t.leftChild = t;
t.rightChild = null;
return t;
}
// Search the key ele
lastNode = search(t, ele);
// If already present key
if (ele == lastNode.data) {
System.out.println(
"Error : key is already present\n");
return t;
}
for (i = 1; bit(ele, i) == bit(lastNode.data, i); i++)
;
current = t.leftChild;
parent = t;
while (current.bitNumber > parent.bitNumber
&& current.bitNumber < i) {
parent = current;
current = (bit(ele, current.bitNumber))
? current.rightChild
: current.leftChild;
}
newNode = new CompressedNode();
newNode.bitNumber = i;
newNode.data = ele;
newNode.leftChild = bit(ele, i) ? current : newNode;
newNode.rightChild = bit(ele, i) ? newNode : current;
if (current == parent.leftChild)
parent.leftChild = newNode;
else
parent.rightChild = newNode;
return t;
}
Java
// Java program to implement the
// Compressed Trie
class Trie {
// Root Node
private final Node root = new Node(false);
// 'a' for lower, 'A' for upper
private final char CASE;
// Default case
public Trie() { CASE = 'a'; }
// Constructor accepting the
// starting symbol
public Trie(char CASE)
{
this.CASE = CASE;
}
// Function to insert a word in
// the compressed trie
public void insert(String word)
{
// Store the root
Node trav = root;
int i = 0;
// Iterate i less than word
// length
while (i < word.length()
&& trav.edgeLabel[word.charAt(i) - CASE]
!= null) {
// Find the index
int index = word.charAt(i) - CASE, j = 0;
StringBuilder label = trav.edgeLabel[index];
// Iterate till j is less
// than label length
while (j < label.length() && i < word.length()
&& label.charAt(j) == word.charAt(i)) {
++i;
++j;
}
// If is the same as the
// label length
if (j == label.length()) {
trav = trav.children[index];
}
else {
// Inserting a prefix of
// the existing word
if (i == word.length()) {
Node existingChild
= trav.children[index];
Node newChild = new Node(true);
StringBuilder remainingLabel
= strCopy(label, j);
// Making "faceboook"
// as "face"
label.setLength(j);
// New node for "face"
trav.children[index] = newChild;
newChild
.children[remainingLabel.charAt(0)
- CASE]
= existingChild;
newChild
.edgeLabel[remainingLabel.charAt(0)
- CASE]
= remainingLabel;
}
else {
// Inserting word which has
// a partial match with
// existing word
StringBuilder remainingLabel
= strCopy(label, j);
Node newChild = new Node(false);
StringBuilder remainingWord
= strCopy(word, i);
// Store the trav in
// temp node
Node temp = trav.children[index];
label.setLength(j);
trav.children[index] = newChild;
newChild
.edgeLabel[remainingLabel.charAt(0)
- CASE]
= remainingLabel;
newChild
.children[remainingLabel.charAt(0)
- CASE]
= temp;
newChild
.edgeLabel[remainingWord.charAt(0)
- CASE]
= remainingWord;
newChild
.children[remainingWord.charAt(0)
- CASE]
= new Node(true);
}
return;
}
}
// Insert new node for new word
if (i < word.length()) {
trav.edgeLabel[word.charAt(i) - CASE]
= strCopy(word, i);
trav.children[word.charAt(i) - CASE]
= new Node(true);
}
else {
// Insert "there" when "therein"
// and "thereafter" are existing
trav.isEnd = true;
}
}
// Function that creates new String
// from an existing string starting
// from the given index
private StringBuilder strCopy(
CharSequence str, int index)
{
StringBuilder result
= new StringBuilder(100);
while (index != str.length()) {
result.append(str.charAt(index++));
}
return result;
}
// Function to print the Trie
public void print()
{
printUtil(root, new StringBuilder());
}
// Fuction to print the word
// starting from the given node
private void printUtil(
Node node, StringBuilder str)
{
if (node.isEnd) {
System.out.println(str);
}
for (int i = 0;
i < node.edgeLabel.length; ++i) {
// If edgeLabel is not
// NULL
if (node.edgeLabel[i] != null) {
int length = str.length();
str = str.append(node.edgeLabel[i]);
printUtil(node.children[i], str);
str = str.delete(length, str.length());
}
}
}
// Function to search a word
public boolean search(String word)
{
int i = 0;
// Stores the root
Node trav = root;
while (i < word.length()
&& trav.edgeLabel[word.charAt(i) - CASE]
!= null) {
int index = word.charAt(i) - CASE;
StringBuilder label = trav.edgeLabel[index];
int j = 0;
while (i < word.length()
&& j < label.length()) {
// Character mismatch
if (word.charAt(i) != label.charAt(j)) {
return false;
}
++i;
++j;
}
if (j == label.length() && i <= word.length()) {
// Traverse further
trav = trav.children[index];
}
else {
// Edge label is larger
// than target word
// searching for "face"
// when tree has "facebook"
return false;
}
}
// Target word fully traversed
// and current node is word
return i == word.length() && trav.isEnd;
}
// Function to search the prefix
public boolean startsWith(String prefix)
{
int i = 0;
// Stores the root
Node trav = root;
while (i < prefix.length()
&& trav.edgeLabel[prefix.charAt(i) - CASE]
!= null) {
int index = prefix.charAt(i) - CASE;
StringBuilder label = trav.edgeLabel[index];
int j = 0;
while (i < prefix.length()
&& j < label.length()) {
// Character mismatch
if (prefix.charAt(i) != label.charAt(j)) {
return false;
}
++i;
++j;
}
if (j == label.length()
&& i <= prefix.length()) {
// Traverse further
trav = trav.children[index];
}
else {
// Edge label is larger
// than target word,
// which is fine
return true;
}
}
return i == prefix.length();
}
}
// Node class
class Node {
// Number of symbols
private final static int SYMBOLS = 26;
Node[] children = new Node[SYMBOLS];
StringBuilder[] edgeLabel = new StringBuilder[SYMBOLS];
boolean isEnd;
// Function to check if the end
// of the string is reached
public Node(boolean isEnd)
{
this.isEnd = isEnd;
}
}
class GFG {
// Driver Code
public static void main(String[] args)
{
Trie trie = new Trie();
// Insert words
trie.insert("facebook");
trie.insert("face");
trie.insert("this");
trie.insert("there");
trie.insert("then");
// Print inserted words
trie.print();
// Check if these words
// are present or not
System.out.println(
trie.search("there"));
System.out.println(
trie.search("therein"));
System.out.println(
trie.startsWith("th"));
System.out.println(
trie.startsWith("fab"));
}
}
但是对于压缩的 trie,树的重新设计将如下所示,在一般的 trie 中,边 ‘a’ 由引用数组中的这个特定元素表示,但在压缩的 trie 中,“边 ‘face’ 是由引用数组中的这个特定元素表示”。代码是-:
Java
class node {
node[] children = new node[26];
StringBuilder[] edgeLabel = new StringBuilder[26];
boolean isEnd;
}
压缩树中的节点:
Java
class CompressedNode {
int bitNumber;
int data;
CompressedNode leftChild, rightChild;
}
类压缩树:
Java
class CompressedNode {
// Root Node
private CompressedNode root;
private static final int MaxBits = 10;
// Constructor
public CompressedNode() { root = null; }
// Function to check if empty
public boolean isEmpty() { return root == null; }
// Function to clear
public void makeEmpty() { root = null; }
}
在压缩树中搜索:
在压缩的 Trie 树中搜索很像搜索。在这里,我们不是比较单个字符,而是比较字符串。
Java
// Function to search a key k
// in the trie
public boolean search(int k)
{
// Find the number of bits
int numOfBits = (int)(Math.log(k) / Math.log(2)) + 1;
// If error occurs
if (numOfBits > MaxBits) {
System.out.println("Error : Number too large");
return false;
}
// Search Node
CompressedNode searchNode = search(root, k);
// If the data matches
if (searchNode.data == k)
return true;
// Else return false
else
return false;
}
在压缩树中插入一个元素:
Java
// Function to implement the insert
// functionality in the trie
private CompressedNode insert(
CompressedNode t, int ele)
{
CompressedNode current, parent;
CompressedNode lastNode, newNode;
int i;
// If Node is NULL
if (t == null) {
t = new CompressedNode();
t.bitNumber = 0;
t.data = ele;
t.leftChild = t;
t.rightChild = null;
return t;
}
// Search the key ele
lastNode = search(t, ele);
// If already present key
if (ele == lastNode.data) {
System.out.println(
"Error : key is already present\n");
return t;
}
for (i = 1; bit(ele, i) == bit(lastNode.data, i); i++)
;
current = t.leftChild;
parent = t;
while (current.bitNumber > parent.bitNumber
&& current.bitNumber < i) {
parent = current;
current = (bit(ele, current.bitNumber))
? current.rightChild
: current.leftChild;
}
newNode = new CompressedNode();
newNode.bitNumber = i;
newNode.data = ele;
newNode.leftChild = bit(ele, i) ? current : newNode;
newNode.rightChild = bit(ele, i) ? newNode : current;
if (current == parent.leftChild)
parent.leftChild = newNode;
else
parent.rightChild = newNode;
return t;
}
下面是实现压缩 Trie 的所有功能的程序:
Java
// Java program to implement the
// Compressed Trie
class Trie {
// Root Node
private final Node root = new Node(false);
// 'a' for lower, 'A' for upper
private final char CASE;
// Default case
public Trie() { CASE = 'a'; }
// Constructor accepting the
// starting symbol
public Trie(char CASE)
{
this.CASE = CASE;
}
// Function to insert a word in
// the compressed trie
public void insert(String word)
{
// Store the root
Node trav = root;
int i = 0;
// Iterate i less than word
// length
while (i < word.length()
&& trav.edgeLabel[word.charAt(i) - CASE]
!= null) {
// Find the index
int index = word.charAt(i) - CASE, j = 0;
StringBuilder label = trav.edgeLabel[index];
// Iterate till j is less
// than label length
while (j < label.length() && i < word.length()
&& label.charAt(j) == word.charAt(i)) {
++i;
++j;
}
// If is the same as the
// label length
if (j == label.length()) {
trav = trav.children[index];
}
else {
// Inserting a prefix of
// the existing word
if (i == word.length()) {
Node existingChild
= trav.children[index];
Node newChild = new Node(true);
StringBuilder remainingLabel
= strCopy(label, j);
// Making "faceboook"
// as "face"
label.setLength(j);
// New node for "face"
trav.children[index] = newChild;
newChild
.children[remainingLabel.charAt(0)
- CASE]
= existingChild;
newChild
.edgeLabel[remainingLabel.charAt(0)
- CASE]
= remainingLabel;
}
else {
// Inserting word which has
// a partial match with
// existing word
StringBuilder remainingLabel
= strCopy(label, j);
Node newChild = new Node(false);
StringBuilder remainingWord
= strCopy(word, i);
// Store the trav in
// temp node
Node temp = trav.children[index];
label.setLength(j);
trav.children[index] = newChild;
newChild
.edgeLabel[remainingLabel.charAt(0)
- CASE]
= remainingLabel;
newChild
.children[remainingLabel.charAt(0)
- CASE]
= temp;
newChild
.edgeLabel[remainingWord.charAt(0)
- CASE]
= remainingWord;
newChild
.children[remainingWord.charAt(0)
- CASE]
= new Node(true);
}
return;
}
}
// Insert new node for new word
if (i < word.length()) {
trav.edgeLabel[word.charAt(i) - CASE]
= strCopy(word, i);
trav.children[word.charAt(i) - CASE]
= new Node(true);
}
else {
// Insert "there" when "therein"
// and "thereafter" are existing
trav.isEnd = true;
}
}
// Function that creates new String
// from an existing string starting
// from the given index
private StringBuilder strCopy(
CharSequence str, int index)
{
StringBuilder result
= new StringBuilder(100);
while (index != str.length()) {
result.append(str.charAt(index++));
}
return result;
}
// Function to print the Trie
public void print()
{
printUtil(root, new StringBuilder());
}
// Fuction to print the word
// starting from the given node
private void printUtil(
Node node, StringBuilder str)
{
if (node.isEnd) {
System.out.println(str);
}
for (int i = 0;
i < node.edgeLabel.length; ++i) {
// If edgeLabel is not
// NULL
if (node.edgeLabel[i] != null) {
int length = str.length();
str = str.append(node.edgeLabel[i]);
printUtil(node.children[i], str);
str = str.delete(length, str.length());
}
}
}
// Function to search a word
public boolean search(String word)
{
int i = 0;
// Stores the root
Node trav = root;
while (i < word.length()
&& trav.edgeLabel[word.charAt(i) - CASE]
!= null) {
int index = word.charAt(i) - CASE;
StringBuilder label = trav.edgeLabel[index];
int j = 0;
while (i < word.length()
&& j < label.length()) {
// Character mismatch
if (word.charAt(i) != label.charAt(j)) {
return false;
}
++i;
++j;
}
if (j == label.length() && i <= word.length()) {
// Traverse further
trav = trav.children[index];
}
else {
// Edge label is larger
// than target word
// searching for "face"
// when tree has "facebook"
return false;
}
}
// Target word fully traversed
// and current node is word
return i == word.length() && trav.isEnd;
}
// Function to search the prefix
public boolean startsWith(String prefix)
{
int i = 0;
// Stores the root
Node trav = root;
while (i < prefix.length()
&& trav.edgeLabel[prefix.charAt(i) - CASE]
!= null) {
int index = prefix.charAt(i) - CASE;
StringBuilder label = trav.edgeLabel[index];
int j = 0;
while (i < prefix.length()
&& j < label.length()) {
// Character mismatch
if (prefix.charAt(i) != label.charAt(j)) {
return false;
}
++i;
++j;
}
if (j == label.length()
&& i <= prefix.length()) {
// Traverse further
trav = trav.children[index];
}
else {
// Edge label is larger
// than target word,
// which is fine
return true;
}
}
return i == prefix.length();
}
}
// Node class
class Node {
// Number of symbols
private final static int SYMBOLS = 26;
Node[] children = new Node[SYMBOLS];
StringBuilder[] edgeLabel = new StringBuilder[SYMBOLS];
boolean isEnd;
// Function to check if the end
// of the string is reached
public Node(boolean isEnd)
{
this.isEnd = isEnd;
}
}
class GFG {
// Driver Code
public static void main(String[] args)
{
Trie trie = new Trie();
// Insert words
trie.insert("facebook");
trie.insert("face");
trie.insert("this");
trie.insert("there");
trie.insert("then");
// Print inserted words
trie.print();
// Check if these words
// are present or not
System.out.println(
trie.search("there"));
System.out.println(
trie.search("therein"));
System.out.println(
trie.startsWith("th"));
System.out.println(
trie.startsWith("fab"));
}
}
输出:
face
facebook
then
there
this
true
false
true
false
如果您想与行业专家一起参加直播课程,请参阅Geeks Classes Live