function memoize(fn) {
  const cache = {};
  return (arg) => cache[arg] ?? (cache[arg] = fn(arg));
}

const square = memoize((n) => n * n);

console.log(square(5)); 
console.log(square(5)); 
console.log(square(6));