diff --git a/general/ex4.py b/general/ex4.py new file mode 100755 index 0000000..d86d5ae --- /dev/null +++ b/general/ex4.py @@ -0,0 +1,19 @@ +#!/usr/bin/env python + + +def naive_solution(x: int) -> int: + assert 0 <= x < 10, f"{x} is not a digit" + xx = x * 10 + x + xxx = xx * 10 + x + xxxx = xxx * 10 + x + return x + xx + xxx + xxxx + + +def solution(x: int) -> int: + assert 0 <= x < 10, f"{x} is not a digit" + return x * 1234 + + +if __name__ == "__main__": + for x in range(10): + print(f"{naive_solution(x)} == {solution(x)}")