From c58936690dfbbd4fd2663398eb5f1ae95d062fcc Mon Sep 17 00:00:00 2001 From: bumpsoo Date: Wed, 5 Mar 2025 13:05:16 +0000 Subject: [PATCH] https://leetcode.com/problems/count-total-number-of-colored-cells --- medium/count_total_number_of_colored_cells.py | 8 ++++++++ 1 file changed, 8 insertions(+) create mode 100644 medium/count_total_number_of_colored_cells.py diff --git a/medium/count_total_number_of_colored_cells.py b/medium/count_total_number_of_colored_cells.py new file mode 100644 index 0000000..16f115e --- /dev/null +++ b/medium/count_total_number_of_colored_cells.py @@ -0,0 +1,8 @@ +# https://leetcode.com/problems/count-total-number-of-colored-cells + +class Solution: + def coloredCells(self, n: int) -> int: + if n == 1: + return 1 + return (n - 1) ** 2 + n ** 2 +