JavaScript 从浅入深 100 题训练计划--(18)函数缓存（Memoization）

题目：实现一个 memoize 函数，缓存函数的计算结果

const fib = n => n <= 1 ? n : fib(n-1) + fib(n-2);
const memoizedFib = memoize(fib);

 

const fib = n => n <= 1 ? n : fib(n-1) + fib(n-2)

function memoize(fn, keyGenerator) {
  const cache = new Map();
  
  return function(...args) {
    // 生成缓存 key
    const key = keyGenerator 
      ? keyGenerator(...args) 
      : JSON.stringify(args);
    
    // 检查缓存
    if (cache.has(key)) {
      return cache.get(key);
    }
    
    // 计算并缓存结果
    const result = fn.apply(this, args);
    cache.set(key, result);
    
    return result;
  };
}

const fibM = (function(){
  const cache = new Map();

   function fn(n){
    if(n < 1) return n;
    if(cache.has(n)) return cache.get(n)

    const result = fn(n - 1) +fn(n -2)
    cache.set(n,result)
  }

  return fn
})();


console.time('Without memoization');
fib(40); // 非常慢
console.timeEnd('Without memoization');

const memoizedFib = memoize(fib);
console.time('With memoization');
memoizedFib(40); //  -- 实际也很慢
console.timeEnd('With memoization');

console.time('With fibM');
fibM(40); // 很快
console.timeEnd('With fibM');