Untitled

func coloredCells(n int) int64 {
    var ans int64 = 1

    var somantor int64 = int64(n*(n-1)/2)

    ans = ans + (somantor * 4)

    return ans
}

//ans = 1 + 4(1 + 2 + 3 + 4)