#!/bin/python3

def superDigit(n, k):
    p = create_p(n, k)
    return get_super_digit(p)

def get_super_digit(p):
    if len(p) == 1:
        return int(p)
    else:
        digits = map(int, list(p))
        return get_super_digit(str(sum(digits)))

def create_p(n, k):
    return n * k

if __name__ == '__main__':
    nk = input().split()
    n = nk[0]
    k = int(nk[1])
    result = superDigit(n, k)
    print(result)

@@ -14,25 +14,26 @@
 3%0A%0Adef super
-D
+_d
 igit(n, k):%0A
@@ -40,25 +40,33 @@
     
-p = create_p(n, k
+digits = map(int, list(n)
 )%0A  
@@ -90,17 +90,36 @@
 r_digit(
-p
+str(sum(digits) * k)
 )%0A%0Adef g
@@ -278,46 +278,8 @@
 ))%0A%0A
-def create_p(n, k):%0A    return n * k%0A%0A
 if _
@@ -381,9 +381,10 @@
 uper
-D
+_d
 igit

Implement shortcut to compute initial p super digit