class TextSearch: def init(self, files): self.files = files self.index = {}

    for file_name, words in files.items():
        for i, word in enumerate(words):
            if word in self.index:
                self.index[word].append((file_name, i))
            else:
                self.index[word] = [(file_name, i)]

def find(self, word):
    result = {}
    if word in self.index:
        occurrences = self.index[word]
        for file_name, position in occurrences:
            if file_name in result:
                continue
            result[file_name] = position
    return result

def count(self, word):
    result = {}
    if word in self.index:
        occurrences = self.index[word]
        for file_name, _ in occurrences:
            if file_name in result:
                result[file_name] += 1
            else:
                result[file_name] = 1
    return result

Пример использования

files = { 'file1.txt': ['apple', 'banana', 'apple', 'orange'], 'file2.txt': ['apple', 'orange', 'grape', 'apple'] }

search = TextSearch(files)

print(search.find('apple')) # {'file1.txt': 0, 'file2.txt': 0} print(search.count('apple')) # {'file1.txt': 2, 'file2.txt': 2}