2579-count-total-number-of-colored-cells.py

class Solution:
    def coloredCells(self, n: int) -> int:
        return n*n + (n-1)*(n-1)