mirror of
https://github.com/TheAlgorithms/Python.git
synced 2025-07-07 03:07:46 +08:00
Implemented Suffix Tree Data Structure (#11554)
* Implemented KD-Tree Data Structure * Implemented KD-Tree Data Structure. updated DIRECTORY.md. * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Create __init__.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Replaced legacy `np.random.rand` call with `np.random.Generator` in kd_tree/example_usage.py * Replaced legacy `np.random.rand` call with `np.random.Generator` in kd_tree/hypercube_points.py * added typehints and docstrings * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * docstring for search() * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Added tests. Updated docstrings/typehints * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * updated tests and used | for type annotations * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * E501 for build_kdtree.py, hypercube_points.py, nearest_neighbour_search.py * I001 for example_usage.py and test_kdtree.py * I001 for example_usage.py and test_kdtree.py * Update data_structures/kd_tree/build_kdtree.py Co-authored-by: Christian Clauss <cclauss@me.com> * Update data_structures/kd_tree/example/hypercube_points.py Co-authored-by: Christian Clauss <cclauss@me.com> * Update data_structures/kd_tree/example/hypercube_points.py Co-authored-by: Christian Clauss <cclauss@me.com> * Added new test cases requested in Review. Refactored the test_build_kdtree() to include various checks. * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Considered ruff errors * Considered ruff errors * Apply suggestions from code review * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update kd_node.py * imported annotations from __future__ * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Implementation of the suffix tree data structure * Adding data to DIRECTORY.md * Minor file renaming * minor correction * renaming in DIRECTORY.md * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Considering ruff part-1 * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Considering ruff part-2 * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Considering ruff part-3 * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Considering ruff part-4 * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Considering ruff part-5 * Implemented Suffix Tree Data Structure. Added some comments to my files in #11532, #11554. * updating DIRECTORY.md * Implemented Suffix Tree Data Structure. Added some comments to my files in #11532, #11554. --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Christian Clauss <cclauss@me.com> Co-authored-by: Ramy-Badr-Ahmed <Ramy-Badr-Ahmed@users.noreply.github.com>
This commit is contained in:
66
data_structures/suffix_tree/suffix_tree.py
Normal file
66
data_structures/suffix_tree/suffix_tree.py
Normal file
@ -0,0 +1,66 @@
|
||||
# Created by: Ramy-Badr-Ahmed (https://github.com/Ramy-Badr-Ahmed)
|
||||
# in Pull Request: #11554
|
||||
# https://github.com/TheAlgorithms/Python/pull/11554
|
||||
#
|
||||
# Please mention me (@Ramy-Badr-Ahmed) in any issue or pull request
|
||||
# addressing bugs/corrections to this file.
|
||||
# Thank you!
|
||||
|
||||
from data_structures.suffix_tree.suffix_tree_node import SuffixTreeNode
|
||||
|
||||
|
||||
class SuffixTree:
|
||||
def __init__(self, text: str) -> None:
|
||||
"""
|
||||
Initializes the suffix tree with the given text.
|
||||
|
||||
Args:
|
||||
text (str): The text for which the suffix tree is to be built.
|
||||
"""
|
||||
self.text: str = text
|
||||
self.root: SuffixTreeNode = SuffixTreeNode()
|
||||
self.build_suffix_tree()
|
||||
|
||||
def build_suffix_tree(self) -> None:
|
||||
"""
|
||||
Builds the suffix tree for the given text by adding all suffixes.
|
||||
"""
|
||||
text = self.text
|
||||
n = len(text)
|
||||
for i in range(n):
|
||||
suffix = text[i:]
|
||||
self._add_suffix(suffix, i)
|
||||
|
||||
def _add_suffix(self, suffix: str, index: int) -> None:
|
||||
"""
|
||||
Adds a suffix to the suffix tree.
|
||||
|
||||
Args:
|
||||
suffix (str): The suffix to add.
|
||||
index (int): The starting index of the suffix in the original text.
|
||||
"""
|
||||
node = self.root
|
||||
for char in suffix:
|
||||
if char not in node.children:
|
||||
node.children[char] = SuffixTreeNode()
|
||||
node = node.children[char]
|
||||
node.is_end_of_string = True
|
||||
node.start = index
|
||||
node.end = index + len(suffix) - 1
|
||||
|
||||
def search(self, pattern: str) -> bool:
|
||||
"""
|
||||
Searches for a pattern in the suffix tree.
|
||||
|
||||
Args:
|
||||
pattern (str): The pattern to search for.
|
||||
|
||||
Returns:
|
||||
bool: True if the pattern is found, False otherwise.
|
||||
"""
|
||||
node = self.root
|
||||
for char in pattern:
|
||||
if char not in node.children:
|
||||
return False
|
||||
node = node.children[char]
|
||||
return True
|
Reference in New Issue
Block a user