update 91, 92

python/91.py

@@ -1,25 +1,24 @@
-from sys import argv
 
-n = int(argv[1])
+from tools import number_theory
-total = 3 * n ** 2
 
-if n % 2 == 0:
+def cnt_side(limit):
-  total += n
+    return limit ** 2 * 3
 
-total += (n - 1) / 2 * n
+def cnt_cross(limit):
+    return (limit - 1) // 2 * limit + (1 - limit % 2) * limit
 
-def gcd(x, y):
+def cnt_inner(limit):
-  if y == 0:
+    tale = 0
-    return x
+    for y in range(1, limit - 1):
-  return gcd(y, x % y)
+        for x in range(y + 1, limit + 1):
+            d = number_theory.gcd(x, y)
+            dx = x // d
+            dy = y // d
+            tale += min((limit - x) // dy, y // dx)
+            tale += min((limit - y) // dx, x // dy)
+    return 2 * tale
 
-for i in xrange(1, n - 1):
+def count(limit):
-  for j in xrange(i + 1, n + 1):
+    return cnt_side(limit) + cnt_cross(limit) + cnt_inner(limit)
-    d = gcd(i, j)
-    i_ = i / d
-    j_ = j / d
-    total += min((n - i) / j_, j / i_) * 2
-    total += min((n - j) / i_, i / j_) * 2
 
-
+print(count(50))
-print total

python/92.py

@@ -1,58 +1,54 @@
-maxx = 7
+import time
+from functools import reduce
+from tools import algebra
 
-def make_iter(x, a, l, lis):
+def jump(num):
-  for i in xrange(a % 10, 10):
+    if 9 > num % 10:
-    if l == x - 1:
+        return 1
-      lis.append(a * 10 + i)
     else:
-      make_iter(x, a * 10 + i, l + 1, lis)
+        num += 1
+        offset = 0
+        while not num % 10:
+            num //= 10
+            offset += 1
+        return 1 + (10 ** offset - 1) // 9 * (num % 10)
 
-def make(x):
+def gen_pick(digit):
-  if x == 1:
+    maxi = 10 ** (digit + 1) - 1
-    return [1,2,3,4,5,6,7,8,9]
+    num = maxi // 9
-  out = []
+    while num < maxi:
-  for i in xrange(1, 10):
+        yield num
-    make_iter(x - 1, i, 0, out)
+        num += jump(num)
-  return out
+    yield maxi
 
-#print make(5)
+def judge(num):
+    while True:
+        a = 0
+        while num != 0:
+            a += (num % 10) ** 2
+            num //= 10
+        if a == 1:
+            return False
+        elif a == 89:
+            return True
+        else:
+            num = a
 
-def calc(x):
+def count(limit, s_num):
-  while 1:
+    total = 1
-    a = 0
+    for i in range(len(s_num)):
-    while x != 0:
+        total *= (limit - i)
-      a += (x % 10) ** 2
+    for ch in set(list(s_num)):
-      x /= 10
+        total //= algebra.factorial(s_num.count(ch))
-    if a == 1 or a == 89:
+    return total
-      return a
-    else:
-      x = a
 
-#print calc(44)
+def statistic(limit):
+    total = 0
+    for digit in range(limit):
+        for num in gen_pick(digit):
+            if judge(num):
+                total += count(limit, str(num))
+    return total
 
-def fac(x):
+print(statistic(7))
-  out = 1
+print(time.process_time())
-  for i in xrange(2, x + 1):
-    out *= i
-  return out
-
-def alanum(x):
-  x = str(x)
-  out = 1
-  for i in xrange(maxx, maxx - len(x), -1):
-    out *= i
-  for i in set('.'.join(x).split('.')):
-    tmp = x.count(i)
-    if tmp > 1:
-      out /= fac(tmp)
-  return out
-
-#print alanum(113)
-
-count = 0
-for length in xrange(1, maxx + 1):
-  tmp = make(length)
-  for xx in tmp:
-    if calc(xx) == 89:
-      count += alanum(xx)
-print count

python/tools/algebra.py

@@ -0,0 +1,6 @@
+
+def factorial(n, multi=1):
+    if 1 == n:
+        return multi
+    else:
+        return factorial(n - 1, multi * n)