← Back
Editing: py.cpython-36.pyc
3 �Pf� � @ sL d dl mZmZmZ d dlmZ d dlmZ ddlm Z G dd� de �Z dS ) � )�absolute_import�division�unicode_literals)� text_type)� bisect_left� )�Triec @ sF e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zdd d �Zdd� Z d S )r c C sJ t dd� |j� D ��s td��|| _t|j� �| _d| _dt|�f| _d S )Nc s s | ]}t |t�V qd S )N)� isinstancer )�.0�x� r �/usr/lib/python3.6/py.py� <genexpr> s z Trie.__init__.<locals>.<genexpr>zAll keys must be strings� r ) �all�keys� TypeError�_data�sorted�_keys� _cachestr�len� _cachepoints)�self�datar r r �__init__ s z Trie.__init__c C s || j kS )N)r )r �keyr r r � __contains__ s zTrie.__contains__c C s t | j�S )N)r r )r r r r �__len__ s z Trie.__len__c C s t | j�S )N)�iterr )r r r r �__iter__ s z Trie.__iter__c C s | j | S )N)r )r r r r r � __getitem__ s zTrie.__getitem__Nc C s� |d ks|dks| j r"t| j �S |j| j�rN| j\}}t| j |||� }}nt| j |� }}t� }|t| j �krv|S x,| j | j|�r�|j| j | � |d7 }qxW || _||f| _|S )Nr r )r �set� startswithr r r r �add)r �prefix�lo�hi�start�ir r r r r s z Trie.keysc C sd || j krdS |j| j�r6| j\}}t| j|||�}n t| j|�}|t| j�krTdS | j| j|�S )NTF)r r# r r r r r )r r% r& r' r) r r r �has_keys_with_prefix6 s zTrie.has_keys_with_prefix)N) �__name__� __module__� __qualname__r r r r r! r r* r r r r r s r N) Z __future__r r r Zpip._vendor.sixr Zbisectr Z_baser ZABCTrier r r r �<module> s
Save File
Cancel