got the gravity down, have a bug to fix

1 files changed, 13 insertions, 2 deletions

diff --git a/2010/1a/a.py b/2010/1a/a.py
index 75472fe..e96a53e 100644
--- a/2010/1a/a.py
+++ b/2010/1a/a.py
@@ -6,14 +6,25 @@ def rotate(m):
     n = len(m)
     return [''.join([m[j][i] for j in range(n)[::-1]]) for i in range(n)]
 
+# drop a single row
+def droprow(r):
+    n = len(r)
+    for i in range(n)[::-1]:
+        if r[i] != '.':
+            return r[i+1:]+r[:i+1]
+    return r
+
+# drop the rotated matrix
+def gravity(m):
+    return map(droprow, m)
+
 t = int(f.readline().strip())
 
 for i in range(t):
     (n, k) = map(int, f.readline().strip().split(' '))
     
     mat = [f.readline().strip() for j in range(n)]
-    
-    res = rotate(mat)
+    res = rotate(gravity(mat))
     
     s = "Case #%d: \n%s\n" % (i+1, '\n'.join(res))
     print s