From 0e77307a7e7b83603145167dc285ed1f9da650d0 Mon Sep 17 00:00:00 2001 From: Yuval Adam Date: Sun, 17 Apr 2011 23:13:01 +0300 Subject: fixed the diagonals --- 2010/1a/a.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/2010/1a/a.py b/2010/1a/a.py index 22b1de4..6a8dd80 100644 --- a/2010/1a/a.py +++ b/2010/1a/a.py @@ -8,16 +8,17 @@ def cols(m): def diag(m): n = len(m) d = [zip(range(i+1) ,range(i+1)[::-1]) for i in range(n)] + d += [zip(range(i+1, n) ,range(i+1, n)[::-1]) for i in range(n-1)] print d return [''.join([m[i][j] for (i,j) in diag]) for diag in d] - + + def diag_inv(m): n = len(m) d = [zip(range(i+1),range(i+1)[::-1]) for i in range(n)] + d += [zip(range(i+1, n) ,range(i+1, n)[::-1]) for i in range(n-1)] return [''.join([m[n-1-i][j] for (i,j) in diag]) for diag in d] -# still need the bottom triangles!!!!! ^^^^^ - def rotate(m): n = len(m) return [''.join([m[j][i] for j in range(n)[::-1]]) for i in range(n)] -- cgit v1.3.1