diff options
Diffstat (limited to 'count_total_coloured_cell.py')
-rw-r--r-- | count_total_coloured_cell.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/count_total_coloured_cell.py b/count_total_coloured_cell.py new file mode 100644 index 0000000..bcff9a9 --- /dev/null +++ b/count_total_coloured_cell.py @@ -0,0 +1,24 @@ +# 2579. Count Total Number of Colored Cells + +""" +there exists an infinitely large two-dimensional grid of uncoloured unit +cells. you are given a positive integer 'n', indicating that you must do the +following routine for 'n' minutes: at the first minute, colour any arbitrary +unit cell blue, and every minute thereafter, colour blue every uncoloured +cell that touches a blue cell. +""" + + +class Solution(object): + def coloredCells(self, n): + """ + :type n: int + :rtype: int + """ + return 1 + 4 * sum(range(n)) + + +if __name__ == "__main__": + obj = Solution() + print(obj.coloredCells(n=1)) + print(obj.coloredCells(n=2)) |