From 7a773c1f45bbecaa8775e3fa6d0e0b4221c899a7 Mon Sep 17 00:00:00 2001
From: Shlomi Fish <shlomif@shlomifish.org>
Date: Wed, 8 Nov 2017 18:16:08 +0200
Subject: [PATCH] Cleanup / optimize same-digits 052.

---
 052/same-digits.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/052/same-digits.py b/052/same-digits.py
index 8ac8f8a..ae81f82 100644
--- a/052/same-digits.py
+++ b/052/same-digits.py
@@ -5,8 +5,10 @@ def get_digits(n):
 
 def same_digits(n, iterations=2):
     digits = get_digits(n)
+    prod = n
     for i in range(1, iterations):
-        if not digits == get_digits(n+(i*n)):
+        prod += n
+        if not digits == get_digits(prod):
             return False
     return True