dost

@@ -9,7 +9,7 @@ from typing import Generator

 def cost(token_child: tuple[str, "Trie"]) -> tuple:

     token, child = token_child

     weight = child.deep_weight

-    return child.deep_usage / weight, child.min_depth, len(token), token

+    return child.deep_usage / weight, -weight, child.min_depth, len(token), token

 class Trie:

     def __init__(self, weight: int = 0):

@@ -28,7 +28,7 @@ class Trie:

     def add(self, word: list[str], weight: int = 1):

         self.deep_weight += weight

-        self.min_depth = min(self.min_depth, sum(len(t) for t in word))

+        self.min_depth = min(self.min_depth, len(word))

         if word:

             self.children[word[0]].add(word[1:], weight)

         else: