cs_totalSquares

assessment

by David McClelland

JavaScript

function solution(n) {
  // the number of squares is (n * 4) + (n-1 * 4) + ... + 1
  // the test does not include n = 1 as a condition
  let totalSquares = 1;
  for (let i = 1; i < n; i++) {
      totalSquares += (n - i) * 4;
  }
  return totalSquares;
}

console.log(solution(2));