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� s z Trie.__init__..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
� s