Search Suggestions System Python Solution

lcpy gen -n 1268   # by problem number
lcpy gen -s search_suggestions_system   # by problem name

Input: products = ["mobile","mouse","moneypot","monitor","mousepad"], searchWord = "mouse"
Output: [["mobile","moneypot","monitor"],["mobile","moneypot","monitor"],["mouse","mousepad"],["mouse","mousepad"],["mouse","mousepad"]]
Explanation: products sorted lexicographically = ["mobile","moneypot","monitor","mouse","mousepad"].
After typing m and mo all products match and we show user ["mobile","moneypot","monitor"].
After typing mou, mous and mouse the system suggests ["mouse","mousepad"].

Input: products = ["havana"], searchWord = "havana"
Output: [["havana"],["havana"],["havana"],["havana"],["havana"],["havana"]]
Explanation: The only word "havana" will be always suggested while typing the search word.

from bisect import bisect_left


class Solution:
    def suggested_products(self, products: list[str], search_word: str) -> list[list[str]]:
        products = sorted(products)
        result: list[list[str]] = []
        prefix = ""
        for ch in search_word:
            prefix += ch
            start = bisect_left(products, prefix)
            matches = []
            for product in products[start : start + 3]:
                if not product.startswith(prefix):
                    break
                matches.append(product)
            result.append(matches)
        return result