got the gravity down, have a bug to fix

2 files changed, 21 insertions, 10 deletions

diff --git a/2010/1a/1.out b/2010/1a/1.out
index a3707e7..3da5a6f 100644
--- a/2010/1a/1.out
+++ b/2010/1a/1.out
@@ -1,11 +1,11 @@
 Case #1: 
 .......
-R......
-RB.....
-BRBR...
-RBB....
 .......
 .......
+R......
+RB.....
+BRB....
+RBBR...
 Case #2: 
 R.....
 BRRR..
@@ -14,11 +14,11 @@ BR....
 BBB...
 BRBR..
 Case #3: 
-BBBR
-RRR.
 ....
 ....
+BBB.
+RRRR
 Case #4: 
-RRB
-BB.
 ...
+RR.
+BBB
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