hihoCoder 1014 Trie树【python版】

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Summer_via/article/details/63281109

python版的简单字典树:

n = int(raw_input())
lis = []
for i in range(n):
    str = raw_input()
    lis.append(str)
trie = {}
for i in range(n):
    p=trie
    for ch in lis[i]:
        if not p.has_key(ch):
            p[ch] = [{}, 0]
        p[ch][1] += 1
        p = p[ch][0]
m = int(raw_input())
for i in range(m):
    str = raw_input()
    p = trie
    ans = 0;
    cnt=0;
    for ch in str:
        if not p.has_key(ch):
            break
        else:
            cnt+=1
            if cnt == len(str):
                ans = p[ch][1]
            p = p[ch][0]
    print ans

 

猜你喜欢

转载自blog.csdn.net/Summer_via/article/details/63281109