题解 | #奶牛的名字管理#

from re import S
#A, 
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param commands string字符串一维数组 
# @param args string字符串二维数组 
# @return string字符串一维数组
#
from typing import List

class TrieNode:
    def __init__(self):
        self.children = {}
        self.is_end_of_word = False

class CowNameSystem:
    def __init__(self):
        self.root = TrieNode()
    def insert(self,name:str) -> None:
        node = self.root
        for char in name:
            if char not in node.children:
                node.children[char] =TrieNode()
            node = node.children[char]
        node.is_end_of_word=True
    def search(self, name: str) -> bool:
        node = self.root
        for char in name:
            if char not in node.children:
                return False
            node = node.children[char]
        return node.is_end_of_word
    
    def startWith(self, name:str) -> bool:
        node = self.root
        for char in name:
            if char not in node.children:
                return False
            node = node.children[char]
        return True

class Solution:
    def cowNameSystem(self, commands: List[str], args: List[List[str]]) -> List[str]:
        result = []
        cow_name_system = CowNameSystem()

        for i, command in enumerate(commands):
            if command == "insert":
                cow_name_system.insert(args[i][0])
                result.append("null")
            elif command == "search":
                found = cow_name_system.search(args[i][0])
                result.append("true" if found else "false") 
            elif command == "startsWith":
                found = cow_name_system.startWith(args[i][0])
                result.append("true" if found else "false")
        return result


# 测试代码
if __name__ == "__main__":
    commands = ["insert", "search", "search", "startsWith", "insert", "search"]
    args = [["apple"], ["apple"], ["app"], ["app"], ["app"], ["app"]]

    solution = Solution()
    result = solution.cowNameSystem(commands,args)
    print(result)


有空来解释