Update recursive_digit_sum.py

Author: sharath4444

recursions/recursive_digit_sum.py

@@ -1,7 +1,7 @@
 """
 The super digit problem is defined as follows:
-Given an integer represented as a string and an integer k,
-the goal is to find the super digit of the number formed by concatenating
+Given an integer n represented as a string and an integer k, 
+the goal is to find the super digit of the number formed by concatenating 
 the integer n k times.
 
 The super digit of a number is defined recursively:
@@ -14,46 +14,44 @@
 
 from __future__ import annotations
 
-
-def superDigit(n: str, k: int) -> int:
+def super_digit(n_str: str, repetitions: int) -> int:
     """
-    Computes the super digit of a number formed by concatenating n k times.
+    Computes the super digit of a number formed by concatenating n_str repetitions times.
 
     Parameters:
-    n (str): The string representation of the integer.
-    k (int): The number of times to concatenate n.
+    n_str (str): The string representation of the integer.
+    repetitions (int): The number of times to concatenate n_str.
 
     Returns:
     int: The super digit of the concatenated number.
 
-    >>> superDigit("148", 3)
+    >>> super_digit("148", 3)
     3
-    >>> superDigit("9875", 4)
+    >>> super_digit("9875", 4)
     8
-    >>> superDigit("123", 3)
+    >>> super_digit("123", 3)
     9
     """
 
-    # Calculate the initial sum of the digits in n
-    digit_sum = sum(int(digit) for digit in n)
-
-    # Multiply the sum by k
-    total_sum = digit_sum * k
-
+    # Calculate the initial sum of the digits in n_str
+    digit_sum = sum(int(digit) for digit in n_str)
+    
+    # Multiply the sum by repetitions
+    total_sum = digit_sum * repetitions
+    
     # Recursive function to find the super digit
     while total_sum >= 10:
         total_sum = sum(int(digit) for digit in str(total_sum))
-
+    
     return total_sum
 
-
-if __name__ == "__main__":
-    # Read input and split it into n and k
+if __name__ == '__main__':
+    # Read input and split it into n_str and repetitions
     first_multiple_input = input().rstrip().split()
-
-    n = first_multiple_input[0]  # n as a string
-    k = int(first_multiple_input[1])  # k as an integer
-
-    # Call the superDigit function and print the result
-    result = superDigit(n, k)
+    
+    n_str = first_multiple_input[0]  # n as a string
+    repetitions = int(first_multiple_input[1])  # repetitions as an integer
+    
+    # Call the super_digit function and print the result
+    result = super_digit(n_str, repetitions)
     print(result)