[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

data_structures/binary_tree/segment_tree_node.py

@@ -1,4 +1,4 @@
-class Node():
+class Node:
     def __init__(self, start, end):
         # Initializes a segment tree node with start and end indices
         self.start = start
@@ -8,17 +8,17 @@ def __init__(self, start, end):
         self.right = None
 
 
-class SegmentTree():
-    def __init__(self, nums, mode='max'):
+class SegmentTree:
+    def __init__(self, nums, mode="max"):
         """
         Initializes the Segment Tree.
         :param nums: List of integers to build the tree from.
         :param mode: Operation mode of the tree ('max' or 'sum').
         """
         self.siz = len(nums)
         self.mode = mode
-        if mode not in {'max', 'sum'}:
-            self.mode = 'max'  # Default to max if invalid mode is given
+        if mode not in {"max", "sum"}:
+            self.mode = "max"  # Default to max if invalid mode is given
 
         # Build the tree from the input list
         self.root = self.build(0, self.siz - 1, nums)
@@ -39,14 +39,14 @@ def build(self, start, end, nums):
             n = Node(start, end)
             n.value = nums[start]
             return n
-        
+
         mid = (start + end) // 2
         root = Node(start, end)
         root.left = self.build(start, mid, nums)
         root.right = self.build(mid + 1, end, nums)
 
         # Set the value according to the mode
-        if self.mode == 'max':
+        if self.mode == "max":
             root.value = max(root.left.value, root.right.value)
         else:
             root.value = root.left.value + root.right.value
@@ -58,24 +58,24 @@ def max_in_range(self, start_index, end_index):
         Queries the maximum value in a given range.
         Only works in 'max' mode.
         """
-        if self.mode == 'sum':
-            raise Exception('Current Segment Tree doesn\'t support finding max') 
-        
+        if self.mode == "sum":
+            raise Exception("Current Segment Tree doesn't support finding max")
+
         if start_index > end_index or start_index < 0 or end_index >= self.siz:
-            raise Exception('Invalid index')
-        
+            raise Exception("Invalid index")
+
         return self.query(self.root, start_index, end_index, 0, self.siz - 1)
 
     def sum_in_range(self, start_index, end_index):
         """
         Queries the sum of values in a given range.
         Only works in 'sum' mode.
         """
-        if self.mode == 'max':
-            raise Exception('Current Segment Tree doesn\'t support summing') 
-        
+        if self.mode == "max":
+            raise Exception("Current Segment Tree doesn't support summing")
+
         if start_index > end_index or start_index < 0 or end_index >= self.siz:
-            raise Exception('Invalid index')
+            raise Exception("Invalid index")
 
         return self.query(self.root, start_index, end_index, 0, self.siz - 1)
 
@@ -92,7 +92,7 @@ def query(self, node, start_index, end_index, start, end):
         # Complete overlap
         if start_index <= start and end <= end_index:
             return node.value
-        
+
         mid = (start + end) // 2
 
         if end_index <= mid:
@@ -103,16 +103,15 @@ def query(self, node, start_index, end_index, start, end):
             return self.query(node.right, start_index, end_index, mid + 1, end)
         else:
             # Range spans both children
-            if self.mode == 'max':
+            if self.mode == "max":
                 return max(
                     self.query(node.left, start_index, end_index, start, mid),
-                    self.query(node.right, start_index, end_index, mid + 1, end)
+                    self.query(node.right, start_index, end_index, mid + 1, end),
                 )
             else:
-                return (
-                    self.query(node.left, start_index, end_index, start, mid) +
-                    self.query(node.right, start_index, end_index, mid + 1, end)
-                )
+                return self.query(
+                    node.left, start_index, end_index, start, mid
+                ) + self.query(node.right, start_index, end_index, mid + 1, end)
 
     def update(self, index, new_value):
         """
@@ -121,8 +120,8 @@ def update(self, index, new_value):
         :param new_value: New value to set.
         """
         if index < 0 or index >= self.siz:
-            raise Exception('Invalid index')
-        
+            raise Exception("Invalid index")
+
         self.modify(self.root, index, new_value, 0, self.siz - 1)
 
     def modify(self, node, index, new_value, start, end):
@@ -137,16 +136,16 @@ def modify(self, node, index, new_value, start, end):
         if start == end:
             node.value = new_value
             return
-        
+
         mid = (start + end) // 2
 
         if index <= mid:
             self.modify(node.left, index, new_value, start, mid)
         else:
             self.modify(node.right, index, new_value, mid + 1, end)
-        
+
         # Recompute current node's value after update
-        if self.mode == 'max':
+        if self.mode == "max":
             node.value = max(node.left.value, node.right.value)
         else:
             node.value = node.left.value + node.right.value
@@ -201,4 +200,4 @@ def modify(self, node, index, new_value, start, end):
 
 st_invalid = SegmentTree(nums, mode='unknown')  # Should default to 'max'
 print(st_invalid.mode)  # Expected: 'max'
-"""
+"""