Count Total Number of Colored Cells

class Solution {
public:
    long long coloredCells(int n) {
        if(n == 1) return 1;
        return 1 + 2LL * n * (n - 1);
    }
};