please optimze it using defautldict like in this one 
"""
from collections import defaultdict
import sys

class Node:
    def __init__(self, name):
        self.name = name
        self.user_id = None                 # ID of the user who locked the node
        self.parent = None                  # Parent node
        self.children = []                  # List of child nodes
        self.is_locked = False              # Lock status of the node
        self.locked_descendant_count = 0    # Number of locked descendants

class Tree:
    def __init__(self, n, m, name_list, queries):
        self.node_list = []
        self.node_map = {}  # Map from node name to Node object

        # Create nodes and map their names
        for name in name_list:
            node = Node(name)
            self.node_list.append(node)
            self.node_map[name] = node

        # Build the tree
        for i in range(n):
            parent = self.node_list[i]
            first_child_index = m * i + 1
            for j in range(m):
                child_index = first_child_index + j
                if child_index < n:
                    child = self.node_list[child_index]
                    parent.children.append(child)
                    child.parent = parent
                else:
                    break

        self.root = self.node_list[0]
        self.user_locked_nodes = defaultdict(set)

        # Process queries
        self.processed_queries = []
        for query in queries:
            self.processed_queries.append(query.strip().split())


    def lock(self, node, user_id):
        # If the node is already locked, return False
        if node.is_locked:
            return False

        # If any ancestors are locked, cannot lock this node
        if self.is_ancestor_locked(node):
            return False

        # If any descendants are locked, cannot lock this node
        if node.locked_descendant_count > 0:
            return False

        # Lock the node
        node.is_locked = True
        node.user_id = user_id
        self.user_locked_nodes[user_id].add(node)

        # Update the locked_descendant_count for ancestors
        parent = node.parent
        while parent:
            parent.locked_descendant_count += 1
            parent = parent.parent

        return True

    def unlock(self, node, user_id):
        # The node must be locked and the user_id must match
        if not node.is_locked or node.user_id != user_id:
            return False

        # Unlock the node
        node.is_locked = False
        node.user_id = None
        self.user_locked_nodes[user_id].remove(node)

        # Update the locked_descendant_count for ancestors
        parent = node.parent
        while parent:
            parent.locked_descendant_count -= 1
            parent = parent.parent

        return True

    def upgrade(self, node, user_id):
        # The node must not be locked
        if node.is_locked:
            return False

        # If any ancestors are locked, cannot upgrade this node
        if self.is_ancestor_locked(node):
            return False

        # The node must have at least one locked descendant
        if node.locked_descendant_count == 0:
            return False

        # Collect locked descendants locked by user_id
        locked_descendants = []
        success = self.collect_locked_descendants(node, user_id, locked_descendants)
        if not success:
            return False

        # Unlock all locked descendants
        for descendant in locked_descendants:
            descendant.is_locked = False
            descendant.user_id = None
            self.user_locked_nodes[user_id].remove(descendant)

            # Update the locked_descendant_count for ancestors
            parent = descendant.parent
            while parent:
                parent.locked_descendant_count -= 1
                parent = parent.parent

        # Lock the current node
        node.is_locked = True
        node.user_id = user_id
        self.user_locked_nodes[user_id].add(node)

        # Update the locked_descendant_count for ancestors
        parent = node.parent
        while parent:
            parent.locked_descendant_count += 1
            parent = parent.parent

        return True

    def collect_locked_descendants(self, node, user_id, locked_descendants):
        for child in node.children:
            if child.is_locked:
                if child.user_id == user_id:
                    locked_descendants.append(child)
                else:
                    return False  # Found a locked node by another user
            elif child.locked_descendant_count > 0:
                if not self.collect_locked_descendants(child, user_id, locked_descendants):
                    return False
        return True

    def is_ancestor_locked(self, node):
        current = node.parent
        while current:
            if current.is_locked:
                return True
            current = current.parent
        return False

    def process_queries(self):
        results = []
        for query in self.processed_queries:
            operation_type, node_name, user_id = query
            node = self.node_map[node_name]
            user_id = int(user_id)
            if operation_type == "1":
                result = self.lock(node, user_id)
            elif operation_type == "2":
                result = self.unlock(node, user_id)
            elif operation_type == "3":
                result = self.upgrade(node, user_id)
            else:
                result = False  # Invalid operation
            results.append('true' if result else 'false')
        return results

def main():
    sys.setrecursionlimit(1 << 25)  # Increase recursion limit if needed
    n = int(sys.stdin.readline())
    m = int(sys.stdin.readline())
    q = int(sys.stdin.readline())

    node_list = []
    for _ in range(n):
        node_list.append(sys.stdin.readline().strip())

    queries = []
    for _ in range(q):
        queries.append(sys.stdin.readline().strip())

    tree = Tree(n, m, node_list, queries)
    results = tree.process_queries()
    for res in results:
        print(res)
main()

"""
for this 
"""
from collections import deque
import sys

class Node:
    def __init__(self, name):
        self.name = name
        self.user_id = None                 # ID of the user who locked the node
        self.parent = None                  # Parent node
        self.children = []                  # List of child nodes
        self.is_locked = False              # Lock status of the node
        self.locked_descendant_count = 0    # Number of locked descendants

class Tree:
    def __init__(self, n, m, name_list):
        self.node_list = []
        self.node_map = {}  # Map from node name to Node object

        # Create nodes and map their names
        for name in name_list:
            node = Node(name)
            self.node_list.append(node)
            self.node_map[name] = node

        # Build the tree
        for i in range(n):
            parent = self.node_list[i]
            first_child_index = m * i + 1
            for j in range(m):
                child_index = first_child_index + j
                if child_index < n:
                    child = self.node_list[child_index]
                    parent.children.append(child)
                    child.parent = parent
                else:
                    break

    def lock(self, node, user_id):
        # If the node is already locked, return False
        if node.is_locked:
            return False

        # If any ancestors are locked, cannot lock this node
        current = node.parent
        while current:
            if current.is_locked:
                return False
            current = current.parent

        # If any descendants are locked, cannot lock this node
        if node.locked_descendant_count > 0:
            return False

        # Lock the node
        node.is_locked = True
        node.user_id = user_id

        # Update the locked_descendant_count for all ancestors
        current = node.parent
        while current:
            current.locked_descendant_count += 1
            current = current.parent

        return True

    def unlock(self, node, user_id):
        # The node must be locked and the user_id must match
        if not node.is_locked or node.user_id != user_id:
            return False

        # Unlock the node
        node.is_locked = False
        node.user_id = None

        # Update the locked_descendant_count for all ancestors
        current = node.parent
        while current:
            current.locked_descendant_count -= 1
            current = current.parent

        return True

    def upgrade(self, node, user_id):
        # The node must not be locked
        if node.is_locked:
            return False

        # If any ancestors are locked, cannot upgrade this node
        current = node.parent
        while current:
            if current.is_locked:
                return False
            current = current.parent

        # The node must have at least one locked descendant
        if node.locked_descendant_count == 0:
            return False

        # Collect locked descendants locked by user_id using iterative DFS
        stack = deque()
        stack.append(node)
        locked_descendants = []
        while stack:
            current_node = stack.pop()
            for child in current_node.children:
                if child.is_locked:
                    if child.user_id == user_id:
                        locked_descendants.append(child)
                    else:
                        return False  # Found a locked node by another user
                elif child.locked_descendant_count > 0:
                    stack.append(child)

        # If no locked descendants found, return False
        if not locked_descendants:
            return False

        # Unlock all locked descendants
        for descendant in locked_descendants:
            descendant.is_locked = False
            descendant.user_id = None
            # Update locked_descendant_count for ancestors
            current = descendant.parent
            while current:
                current.locked_descendant_count -= 1
                current = current.parent

        # Lock the current node
        node.is_locked = True
        node.user_id = user_id

        # Update the locked_descendant_count for ancestors
        current = node.parent
        while current:
            current.locked_descendant_count += 1
            current = current.parent

        return True

    def process_query(self, operation_type, node_name, user_id):
        node = self.node_map[node_name]
        if operation_type == "1":
            return self.lock(node, user_id)
        elif operation_type == "2":
            return self.unlock(node, user_id)
        elif operation_type == "3":
            return self.upgrade(node, user_id)
        else:
            return False  # Invalid operation

def main():
    sys.setrecursionlimit(1 << 25)  # Increase recursion limit if necessary

    # Read input
    n = int(sys.stdin.readline())
    m = int(sys.stdin.readline())
    q = int(sys.stdin.readline())

    name_list = [sys.stdin.readline().strip() for _ in range(n)]
    tree = Tree(n, m, name_list)

    # Process queries
    for _ in range(q):
        query = sys.stdin.readline().strip().split()
        operation_type, node_name, user_id = query
        user_id = int(user_id)
        result = tree.process_query(operation_type, node_name, user_id)
        print('true' if result else 'false')

main()

"""
which is better and do the best and give me a complete code which will pass all tcs and no tles for the problem 
"""
# **Locking The Tree of Space**

You have a world map represented as an M-Ary tree (sample tree below)  

<aside>

<img src="./images/asterisk_green.svg" alt="asterisk_green.svg" width="15px" /> In graph theory, an m-ary tree (for nonnegative integers m) (also known as n-ary, k-ary or k-way tree) is an arborescence (or an ordered tree) in which each node has no more than m children. A binary tree is the special case where m = 2, and a ternary tree is another case with m = 3 that limits its children to three.

<img src="./images/asterisk_green.svg" alt="asterisk_green.svg" width="15px" /> In graph theory, an arborescence is a directed graph in which, for a vertex u (called the root) and any other vertex v, there is exactly one directed path from u to v. A graph is a collection of vertices (also known as nodes) and edges that connect these vertices. A tree is a special type of graph that is connected and acyclic, meaning that there are no cycles in the graph. In a tree, there is a unique path between any two vertices, and there is a single vertex called the root that is used as the starting point for traversing the tree. 

</aside>

![Sample Tree](./images/Sample%20Tree.png)

You need to define three operations on it:

```cpp
1. lock(X, uid)
2. unlock(X, uid)
3. upgradeLock(X, uid)
```

where `X` the name of a node in the tree (that would be unique) and `uid` is the user who is performing the operation.  
**Input format**: x is country name (string) and uid is user number (number).

**Here are the definitions for the operations:**

>**`lock(x, uid)`**  
Lock takes an exclusive access on the subtree rooted at `X`. It is formally defined like this: Once `lock(x, uid)` succeeds, then:  
    <li>`lock(A, anyUserid)` should fail (returns false), where `A` is a descendent of `X`,</li>
    <li>`lock(B, anyUserld)` should fail (returns false), where `X` is a descendent of `B`,</li>
    <li>Lock operation cannot be performed on a node which is already locked i.e. `lock(x, anyUserld)` should fail (returns false).</li>  
    <aside>
    <img src="./images/asterisk_green.svg" alt="asterisk_green.svg" width="15px" /> Exclusive access means that only one user can lock a node and its descendants at a time, and no other user can lock any node that is related to the locked node by ancestry or descent. This ensures that there is no conflict or interference between different users who want to perform operations on the tree.
    </aside> 

>**`unlock(X, uid)`**  
Unlock reverts what was done by the lock operation. It can only be called on same node on which user `uid` had called a Lock on before. Returns true if it is successful.

>**`upgradeLock(x, uid)`**  
It helps the user `uid` upgrade their lock to an ancestor node. It is only possible if the node `X` already has locked descendants and all of them are only locked by the same user `uid`. Upgrade should fail if there is any node which is descendant of X that is locked by a different user. Successful Upgrade will 'lock' the node. `upgradeLock` call shouldn't violate the consistency model that Lock/Unlock function requires.

**Notes**

1. The number of nodes in the tree `N` is very large. So, optimize the time complexity for the above algorithms.
2. The below section contains the input format.
    - The first line contains the **number** of Nodes in the tree (`N`).
    - The second line contains **number** of children per node (value `m` in m-ary Tree).
    - The third line contains **number** of queries (`Q`).
    - Next **`N`** lines contains the **`NodeName`** (string) in the m-Ary Tree.
    - Next **`Q`** lines contains queries which are in format: 
    **`Operation Type` `NodeName` `Userld`**
    - **Operation Type**
        
        `1` for lock  
        `2` for unlock  
        `3` for upgradeLock
        
    - **`NodeName`** - Name of any node (unique) in m-Ary Tree.
    - **`Userld`** - Integer value representing a unique user.

**Example input**

```cpp
7
2
3
World
Asia
Africa
China
India
South Africa
Egypt
1 China 9
2 India 9
3 Asia 9
```

With the above input you represent a 2-ary tree with 7 nodes as follows:

```
                 World
            /             \
        Asia               Africa
      /      \          /         \
China    India    South Africa   Egypt
```

**Additional Notes**

1. Here `1 China 3` indicates the following '`Operation Type` `NodeName` `Userld`'.
2. The tree is always fully balanced.
3. Constraints on the inputs are as follows:  
1 < **N** < 5 * 10^5  
1 < **m** < 30  
1 < **Q** <5 * 10^5  
1 < length of **NodeName** < 20
4. Optimize the time complexity:
    - Lock - O(logmN)
    - Unlock - O(logmN)
    - UpgradeLock - O(numberOfLockedNodes * logmN)
5. Lock operation on already locked node should fail.
6. Once Upgrade lock(X, uid) succeeds on X. It is equivalent to X being locked by uid. So, Lock(A/B, anyuser) should fail as per the definition of Lock and Unlock(X, uid) should also work.
7. Upgrade lock operation on a node having no locked descendants should fail and upgrade lock on already locked node should also fail.

**Example input**

```cpp
7
2
5
World
Asia
Africa
China
India
SouthAfrica
Egypt
1 China 9
1 India 9
3 Asia 9
2 India 9
2 Asia 9
```

**Example input**

```cpp
true
true
true
false
true
```

Query 1: `1 China 9` ⇒ This operation is success as initially China is unlocked. 

Query 2: `1 India 9` ⇒ This should be success as none of ancestors and descendants of India are locked.

Query 3: `3 Asia 9` ⇒ This also should be success as upgrade operation is done by same user who has locked descendants.

Query 4: `2 India 9` ⇒ This should fail as the India is now not locked 

Query 5: `2 Asia 9` ⇒ This should be success as Asia was earlier (refer Query 3) locked by user 9.
"""
plesae understand both codes and give mthe best one modified or inofoed one