TheAlgorithms-Python
32 строки · 645.0 Байт
1"""Multiply two numbers using Karatsuba algorithm"""
2
3
4def karatsuba(a: int, b: int) -> int:
5"""
6>>> karatsuba(15463, 23489) == 15463 * 23489
7True
8>>> karatsuba(3, 9) == 3 * 9
9True
10"""
11if len(str(a)) == 1 or len(str(b)) == 1:
12return a * b
13
14m1 = max(len(str(a)), len(str(b)))
15m2 = m1 // 2
16
17a1, a2 = divmod(a, 10**m2)
18b1, b2 = divmod(b, 10**m2)
19
20x = karatsuba(a2, b2)
21y = karatsuba((a1 + a2), (b1 + b2))
22z = karatsuba(a1, b1)
23
24return (z * 10 ** (2 * m2)) + ((y - z - x) * 10 ** (m2)) + (x)
25
26
27def main():
28print(karatsuba(15463, 23489))
29
30
31if __name__ == "__main__":
32main()
33