Update image-smoother.py

CMWENLIU · web-flow · commit b776b512efc9 · 2018-04-19T15:27:37.000-05:00
Another solution with getGray function simplified.
diff --git a/Python/image-smoother.py b/Python/image-smoother.py
@@ -47,3 +47,25 @@ def getGray(M, i, j):
             for j in xrange(len(M[0])):
                 result[i][j] = getGray(M, i, j);
         return result
+# Another solution with getGray function simplified.    
+class Solution(object):
+    def imageSmoother(self, M):
+        """
+        :type M: List[List[int]]
+        :rtype: List[List[int]]
+        """
+        def getGray(M, i, j):
+            total, count = 0, 0.0
+            for r in xrange(-1, 2):
+                for c in xrange(-1, 2):
+                    ii, jj = i + r, j + c
+                    if 0 <= ii < len(M) and 0 <= jj < len(M[0]):
+                        total += M[ii][jj]
+                        count += 1.0
+            return int(total / count)
+
+        result = [[0 for _ in xrange(len(M[0]))] for _ in xrange(len(M))]
+        for i in xrange(len(M)):
+            for j in xrange(len(M[0])):
+                result[i][j] = getGray(M, i, j);
+        return result
