class WordFilter(object):
def __init__(self, words):
from collections import defaultdict
self.prefixes = defaultdict(set)
self.suffixes = defaultdict(set)
self.weights = {}
for index, word in enumerate(words):
prefix, suffix = '', ''
for char in [''] + list(word):
prefix += char
self.prefixes[prefix].add(word)
for char in [''] + list(word[::-1]):
suffix += char
self.suffixes[suffix[::-1]].add(word)
self.weights[word] = index
def f(self, prefix, suffix):
weight = -1
for word in self.prefixes[prefix] & self.suffixes[suffix]:
if self.weights[word] > weight:
weight = self.weights[word]
return weight
return cur[WEIGHT]
# Your WordFilter object will be instantiated and called as such:
# obj = WordFilter(words)
# param_1 = obj.f(prefix,suffix)
745. Prefix and Suffix Search刷题笔记
最新推荐文章于 2024-07-25 19:34:38 发布