got the rotate function down

1 files changed, 12 insertions, 3 deletions

diff --git a/2010/1a/a.py b/2010/1a/a.py
index cd65727..75472fe 100644
--- a/2010/1a/a.py
+++ b/2010/1a/a.py
@@ -1,11 +1,20 @@
 f = open('1.in', 'r')
 o = open('1.out', 'w')
 
+# rotate *90 degrees clockwise*
+def rotate(m):
+    n = len(m)
+    return [''.join([m[j][i] for j in range(n)[::-1]]) for i in range(n)]
+
 t = int(f.readline().strip())
 
 for i in range(t):
-    nk = f.readline().strip().split(' ')
+    (n, k) = map(int, f.readline().strip().split(' '))
     
-    s = "Case #%d: %s\n" % (i+1, res)
-
+    mat = [f.readline().strip() for j in range(n)]
+    
+    res = rotate(mat)
+    
+    s = "Case #%d: \n%s\n" % (i+1, '\n'.join(res))
+    print s
     o.write(s)