Taddeus Kroes 13 éve
szülő
commit
40eab4bf6a
2 módosított fájl, 9 hozzáadás és 22 törlés
  1. 1 10
      problem29.py
  2. 8 12
      problem81.py

+ 1 - 10
problem29.py

@@ -1,11 +1,2 @@
 #!/usr/bin/env python
-def combos(A, B):
-    c = set()
-
-    for a in A:
-        for b in B:
-            c.add(a ** b)
-
-    return sorted(c)
-
-print len(combos(range(2, 101), range(2, 101)))
+print len(set(a ** b for b in range(2, 101) for a in range(2, 101)))

+ 8 - 12
problem81.py

@@ -1,23 +1,19 @@
 #!/usr/bin/env python
-from numpy import array
-
-m = array([l.split(',') for l in open('matrix.txt', 'r').readlines()],
-          dtype=int)
-h, w = m.shape
+m = [map(int, line.split(',')) for line in open('matrix.txt', 'r').readlines()]
+h, w = len(m), len(m[0])
 
 def add(x, y):
     global m
-
     p = []
-    if x: p.append(m[y, x - 1])
-    if y: p.append(m[y - 1, x])
-    m[y, x] += min(p)
+    if x: p.append(m[y][x - 1])
+    if y: p.append(m[y - 1][x])
+    m[y][x] += min(p)
 
-for i in range(w+h):
-    for x in range(i + 1):
+for i in xrange(w + h):
+    for x in xrange(i + 1):
         y = i - x
 
         if (x or y) and x < w and y < h:
             add(x, y)
 
-print m[h - 1, w - 1]
+print m[h - 1][w - 1]