From e487da3836823186161a05eba847c0d204686846 Mon Sep 17 00:00:00 2001 From: terminaldweller Date: Mon, 18 Dec 2023 23:13:36 -0500 Subject: 661 --- 661/main.py | 40 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 40 insertions(+) create mode 100755 661/main.py (limited to '661') diff --git a/661/main.py b/661/main.py new file mode 100755 index 0000000..d17a8a0 --- /dev/null +++ b/661/main.py @@ -0,0 +1,40 @@ +#!/usr/bin/env python +import typing + + +class Solution: + def getAverage(self, M, i, j): + sum = 0 + count = 0 + for x in range(i - 1, i + 2): + for y in range(j - 1, j + 2): + if x >= 0 and x < len(M) and y >= 0 and y < len(M[x]): + sum += M[x][y] + count += 1 + return sum // count + + def imageSmoother( + self, M: typing.List[typing.List[int]] + ) -> typing.List[typing.List[int]]: + result: typing.List[typing.List[int]] = [ + [0 for _ in range(len(M[0]))] for _ in range(len(M)) + ] + for i in range(len(M)): + for j in range(len(M[i])): + result[i][j] = self.getAverage(M, i, j) + + return result + + +def main(): + img = [[1, 1, 1], [1, 0, 1], [1, 1, 1]] + img2 = [[2, 3, 4], [5, 6, 7], [8, 9, 10], [11, 12, 13]] + img3 = [[100, 200, 100], [200, 50, 200], [100, 200, 100]] + solution = Solution() + print(solution.imageSmoother(img)) + print(solution.imageSmoother(img2)) + print(solution.imageSmoother(img3)) + + +if __name__ == "__main__": + main() -- cgit v1.2.3