終于該好了，該字典樹(shù)能夠?qū)崿F(xiàn)對(duì)于數(shù)據(jù)的模糊匹配。代碼如下：
#!/usr/bin/env python
#-*- coding: UTF-8 -*-
class Node:
def __init__(self):
self.map = {}
self.indexnum = 0
self.endflag = False
def contain(self,key):
return self.map.__contains__(key)
def __getitem__(self,key):
return self.map[key]
def __setitem__(self,key,value):
self.map[key] =value
class TrieTree:
def __init__(self):
self.subNum = 0
self.indexNum = 0
self.subNode = Node()
def add(self,key,trieTree):
self.subNum += 1
self.subNode[key] = trieTree
def __chinese(self,char):
char = unicode(char,"utf8")
buf = []
for word in char:
if word >= u'\u4e00'and word <= u'\u9fa5':
buf.append(word.encode('utf-8'))
elif word == '\n':
#在這里對(duì)索引進(jìn)行標(biāo)記
self.indexNum+=1
return buf
def search(self,buf):
buf = self.__chinese(buf)
current = self
for i in range(len(buf)):
#轉(zhuǎn)化成每個(gè)漢字
#print buf[i]
if current.subNode.contain(buf[i]):
if current.subNode[buf[i]].subNode.endflag == True:
return current.subNode.contain(buf[i])
else:
current = current.subNode[buf[i]]
else:
#如果沒(méi)有匹配上，直接進(jìn)入下一層
continue
def load(self,filename):
try:
sock = open(filename,'r')
buf = sock.read().split('\n')
sock.close()
except IOError:
return None
#讀取每個(gè)詞語(yǔ)
for i in range(len(buf)):
buftmp = self.__chinese(buf[i])
tree = self
#讀取每個(gè)漢字
current = tree
for j in range(len(buftmp)):
if current.subNode.contain(buftmp[j]):
current = current.subNode[buftmp[j]]
else:
sub = TrieTree()
current.add(buftmp[j],sub)
current = sub
if j == len(buftmp) - 1:
current.subNode.endflag = True
current.subNode.indexnum = self.indexNum
if __name__=='__main__':
s = TrieTree()
s.load('citynames')
print s.search('我你')
# s.printSelf()